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'
class Drone(
db.Model
): # creating this class is exactly the same as creating a table in SQL, the stuff inside is make columns for it
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(150))
price = db.Column(db.Integer)
model = db.Column(db.String(150))
user_id = db.Column(db.String, db.ForeignKey('user.token'), nullable=False)
def __init__(self, name, price, model, user_id):
self.name = name
self.price = price
self.model = model
self.user_id = user_id
def __repr__(self):
return f'The following Drone has been added: {self.name} which belongs to {self.user_id}'
def to_dict(self):
return {
"id": self.id,
"name": self.name,
"price": self.price,
"model": self.model
}
class Drone(db.Model):
id = db.Column(db.Integer, primary_key = True)
name = db.Column(db.String(150))
price = db.Column(db.Integer)
model = db.Column(db.String(150))
user_id = db.Column(db.String, db.ForeignKey('user.token'), nullable = False)
def __init__(self,name,price,model,user_id):
self.name = name
self.price = price
self.user_id = user_id
def __repr__(self):
return f'The following Drone has been added: {self.name} which belongs to {self.user_id}'
def to_dict(self):
return{
"id": self.id,
"name": self.name,
"price": self.price,
"model": self.model
}
