class Milkers(db.Model, UserMixin):
__tablename__ = 'milkers'
milker_id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), nullable=False, index=True)
place = db.Column(db.String(64), nullable=False, index=True)
bike = db.Column(db.String(64), nullable=True)
salary = db.Column(db.Float, nullable=False)
#owner_id = db.Column(db.Integer, db.ForeignKey('owners.owner_id'), nullable=True)
owner = db.relationship('CowOwner', backref='milker', lazy='dynamic')
#milk = db.relationship('Milk', backref='owner', lazy=True)
def __init__(self, name, place, bike, salary):
self.name = name
self.place = place
self.bike = bike
self.salary = salary
def __repr__(self):
return f"Milker {self.name} {self.milker_id} from {self.place}"
def milker_customers(self):
return self.owner.all()
class User(db.Model, UserMixin):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
profile_image = db.Column(db.String(64),
nullable=False,
default='default_profile.png')
email = db.Column(db.String(64), unique=True, index=True)
username = db.Column(db.String(20), unique=True, index=True)
password_hash = db.Column(db.String(128))
google = db.relationship('GoogleData', backref='username', lazy=True)
def __init__(self, email, username, password):
self.email = email
self.username = username
print(generate_password_hash(password), "came the password")
self.password_hash = generate_password_hash(password)
print("created user")
def check_password(self, password):
print("Iam here within check pw")
print(password, 'rec pw')
print(self.email, 'stored email')
print(self.username, 'stored uname')
print(self.password_hash, 'pw_hash')
print(self.username, 'user')
print(check_password_hash(self.password_hash, password),
"Pw authenticated successfully")
return check_password_hash(self.password_hash, password)
def __repr__(self):
return f"Username {self.username}"
class GoogleData(db.Model):
__tablename__ = 'Google_data'
users = db.relationship(User)
user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
sheet_name = db.Column(db.String(64),
primary_key=True,
unique=True,
nullable=False)
sheet_link = db.Column(db.Text, unique=True, nullable=False)
date = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
def __init__(self, user_id, sheet_name, sheet_link):
self.user_id = user_id
self.sheet_name = sheet_name
self.sheet_link = sheet_link
def __repr__(self):
return f"Google Sheet Name : {self.sheet_name}"
class CowOwner(db.Model, UserMixin):
__tablename__ = 'owners'
id = db.Column(db.Integer, primary_key=True)
owner_id = db.Column(db.Integer, index=True)
name = db.Column(db.String(64), index=True)
place = db.Column(db.String(64), index=True)
cows = db.Column(db.Integer)
milker_id = db.Column(db.Integer,
db.ForeignKey('milkers.milker_id'),
nullable=False)
#own_cows = db.relationship('Cows', backref='owner', lazy='dynamic')
#milk = db.relationship('Milk', backref='owner', lazy=True)
# milker = db.relationship('Milkers', backref='owner', lazy='dynamic')
def __init__(self, owner_id, name, place, num_cows, milker_id):
self.owner_id = owner_id
self.name = name
self.place = place
self.cows = num_cows
self.milker_id = milker_id
def __repr__(self):
return f"Owner {self.name} place {self.place} num_cows {self.cows} milker {self.milker_id}"
def cows_and_owners(self):
return self.own_cows
class Cows(db.Model):
__tablename__ = 'cows'
cow_id = db.Column(db.Integer, primary_key=True)
breed = db.Column(db.String(64))
owner_id = db.Column(db.Integer,
db.ForeignKey('owners.owner_id'),
nullable=False)
average = db.Column(db.Float, nullable=True)
milker_id = db.Column(db.Integer,
db.ForeignKey('milkers.milker_id'),
nullable=False)
cows = db.relationship('Milkers', backref='milkers', lazy=True)
def __init__(self, breed, owner_id, milker_id):
self.breed = breed
self.owner_id = owner_id
self.milker_id = milker_id
def __repr__(self):
return f"{self.owners.name} is the owner of Cows : {self.cows}"
class Milk(db.Model, UserMixin):
__tablename__ = "milk"
event_id = db.Column(db.Integer,
primary_key=True,
autoincrement=True,
index=True)
#id = db.Column(db.Integer, primary_key=True)
owner_id = db.Column(db.Integer, nullable=False)
milker_id = db.Column(db.Integer,
db.ForeignKey('milkers.milker_id'),
nullable=False,
index=True)
milked_date = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
milked_time = db.Column(db.String(4), nullable=False)
litre = db.Column(db.Integer, nullable=False)
ml = db.Column(db.Integer, nullable=False)
__table_args__ = (UniqueConstraint('event_id',
'owner_id',
'milker_id',
'milked_date',
name='_daily_milk_data'), )
def __init__(self, owner_id, milker, milked_date, milked_time, litre, ml):
self.owner_id = owner_id
#self.name=name
#self.place=place
self.milker_id = milker
self.milked_date = milked_date
self.milked_time = milked_time
self.litre = litre
self.ml = ml
def __repr__(self):
return f"Date: {self.milked_date}"