Create repositories/user_repo.py
Browse files- repositories/user_repo.py +18 -0
repositories/user_repo.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from models.db import get_session
|
2 |
+
from models.user import User
|
3 |
+
from passlib.context import CryptContext
|
4 |
+
|
5 |
+
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
|
6 |
+
|
7 |
+
class UserRepo:
|
8 |
+
def __init__(self, db_url): pass # uses global engine
|
9 |
+
|
10 |
+
def get_all_users(self):
|
11 |
+
with get_session() as s:
|
12 |
+
return s.query(User).all()
|
13 |
+
|
14 |
+
def add_user(self, username, full_name, password):
|
15 |
+
hashed = pwd_context.hash(password)
|
16 |
+
user = User(username=username, full_name=full_name, hashed_password=hashed)
|
17 |
+
with get_session() as s:
|
18 |
+
s.add(user); s.commit()
|