mirror of
https://github.com/Llloooggg/Jarvis.git
synced 2026-03-06 20:16:23 +03:00
28 lines
697 B
Python
28 lines
697 B
Python
from flask_sqlalchemy import SQLAlchemy
|
|
from __init__ import app
|
|
|
|
|
|
db = SQLAlchemy(app)
|
|
|
|
|
|
class User(db.Model):
|
|
__tablename__ = 'users'
|
|
UserID = db.Column(db.Integer, primary_key=True)
|
|
UserName = db.Column(db.String(80), unique=True, nullable=False)
|
|
UserPassw = db.Column(db.String(120), nullable=False)
|
|
|
|
|
|
def add_user(user_name, passw_hash):
|
|
if not find_user(user_name):
|
|
new_user = User(UserName=user_name, UserPassw=passw_hash)
|
|
db.session.add(new_user)
|
|
db.session.commit()
|
|
return True
|
|
else:
|
|
print('Логин занят')
|
|
return False
|
|
|
|
|
|
def find_user(user_name):
|
|
return User.query.filter_by(UserName=user_name).first()
|