backend: регистрация переведена на классы

This commit is contained in:
2022-11-18 15:20:01 +03:00
parent a445fb1578
commit d4a6ad05c0
2 changed files with 25 additions and 23 deletions

View File

@@ -8,33 +8,35 @@ main = Blueprint("main", __name__)
@main.route("/")
def index():
return "Nice!", 200
class Index:
def get(self):
return "Nice!", 200
@main.route("/registration", methods=["POST"])
def registration():
data = request.get_json()
@main.route("/registration")
class Registration:
def post(self):
data = request.get_json()
tg_id = data.get("tg_id")
first_name = data.get("first_name")
last_name = data.get("last_name")
username = data.get("username")
tg_id = data.get("tg_id")
first_name = data.get("first_name")
last_name = data.get("last_name")
username = data.get("username")
existed_user = User.query.filter_by(tg_id=tg_id).first()
existed_user = User.query.filter_by(tg_id=tg_id).first()
if not existed_user:
new_user = User(
tg_id=tg_id,
first_name=first_name,
last_name=last_name,
username=username,
)
if not existed_user:
new_user = User(
tg_id=tg_id,
first_name=first_name,
last_name=last_name,
username=username,
)
db.session.add(new_user)
db.session.commit()
db.session.add(new_user)
db.session.commit()
return {"created": True, "id": new_user.id}
return {"created": True, "id": new_user.id}
else:
return {"created": False, "id": existed_user.id}
else:
return {"created": False, "id": existed_user.id}