class User(db.Model, UserMixin):
id = db.Column(db.String, primary_key = True)
first_name = db.Column(db.String(150), nullable = True, default='')
last_name = db.Column(db.String(150), nullable = True, default = '')
email = db.Column(db.String(150), nullable = False)
password = db.Column(db.String, nullable = True, default = '')
g_auth_verify = db.Column(db.Boolean, default = False)
token = db.Column(db.String, default = '', unique = True)
date_created = db.Column(db.DateTime, nullable = False, default = datetime.utcnow)
drone = db.relationship('Drone', backref = 'owner', lazy = True)
def __init__(self,email,first_name = '', last_name = '', id = '', password = '', token = '', g_auth_verify = False):
self.id = self.set_id()
self.first_name = first_name
self.last_name = last_name
self.password = self.set_password(password)
self.email = email
self.token = self.set_token(24)
self.g_auth_verify = g_auth_verify
def set_token(self,length):
return secrets.token_hex(length)
def set_id(self):
return str(uuid.uuid4())
def set_password(self, password):
self.pw_hash = generate_password_hash(password)
return self.pw_hash
def __repr__(self):
return f'User {self.email} has been added to the database'
class User(db.Model, UserMixin): #order is important. it means take what we have in our model user table as gold. It will find integers first rather than strings first
id = db.Column(db.String, primary_key = True)
first_name = db.Column(db.String(150), nullable = True, default='')
last_name = db.Column(db.String(150), nullable = True, default='')
email = db.Column(db.String(150), nullable = False)
password = db.Column(db.String, nullable = True) #this will let ppl sign in w/ google. So left the nullable to True
g_auth_verify = db.Column(db.Boolean, default = False)
token = db.Column(db.String, default = '', unique = True) #is used for API token. Associate token w/ a length or time restriction (It's unique so that it can be used.)
date_created = db.Column(db.DateTime, nullable = False, default = datetime.utcnow) #False means you dont want the field to be empty
drone = db.relationship('Drone', backref = 'owner', lazy = True)
#in order to use the above, you need to invoke the init method
def __init__(self,email,first_name = '', last_name = '', id = '', password = '', token = '', g_auth_verify = False):
self.id = self.set_id()
self.first_name = first_name
self.last_name = last_name
self.password = self.set_password(password)
self.email = email
self.token = self.set_token(24)
self.g_auth_verify = g_auth_verify
def set_token(self,length):
return secrets.token_hex(length)
def set_id(self):
return str(uuid.uuid4())
def set_password(self, password):
self.pw_hash = generate_password_hash(password)
return self.pw_hash
def __repr__(self):
return f'User {self.email} has been added to the database'
