More database handling and admin interface
This commit is contained in:
15
db/models.py
15
db/models.py
@@ -1,5 +1,5 @@
|
||||
from sqlalchemy import Column, Integer, String
|
||||
from sqlalchemy.orm import declarative_base, sessionmaker
|
||||
from sqlalchemy import Column, Integer, String, Float
|
||||
from sqlalchemy.orm import declarative_base, sessionmaker, Session
|
||||
from sqlalchemy import create_engine
|
||||
from fastapi import Depends
|
||||
|
||||
@@ -17,6 +17,17 @@ class User(Base):
|
||||
id = Column(Integer, primary_key=True)
|
||||
username = Column(String, unique=True, index=True)
|
||||
role = Column(String) # z. B. "admin" oder "user"
|
||||
money = Column(Float, default=0) # money of user
|
||||
|
||||
def create_user(db: Session, username: str, role: str):
|
||||
db_user = User(username=username, role=role)
|
||||
db.add(db_user)
|
||||
db.commit()
|
||||
db.refresh(db_user)
|
||||
return db_user
|
||||
|
||||
def get_user(db: Session, user_id: int):
|
||||
return db.query(User).filter(User.id == user_id).first()
|
||||
|
||||
|
||||
def get_db():
|
||||
|
||||
Reference in New Issue
Block a user