class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True)
email = db.Column(db.String(120), unique=True)
region = db.Column(db.String(100))
password_hash = db.Column(db.String(80))
phone = db.Column(db.String(100))
def __init__(self, email, username, password, phone):
self.email = email
self.username = username
self.password_hash = password
self.phone = phone
@property
def password(self):
raise AttributeError('password: write-only field')
@password.setter
def password(self, password):
password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
@staticmethod
def get_by_username(username):
return User.query.filter_by(username=username).first()
def avatar(self, size):
return 'http://www.gravatar.com/avatar/{}?d=mm&s={}'.format(
md5(self.email.encode('utf-8')).hexdigest(), size)
def __repr__(self):
return '<User %r>' % self.username
class Farmer(db.Model):
id = db.Column(db.Integer, primary_key=True)
number = db.Column(db.String(50), unique=True)
name = db.Column(db.String(50))
surname = db.Column(db.String(50))
location = db.Column(db.String(50))
requests = db.relationship('Question', backref='farmer', lazy='dynamic')
critical_requests = db.relationship('Critical',
backref='farmer',
lazy='dynamic')
unknown_requests = db.relationship('Unknown',
backref='farmer',
lazy='dynamic')
def __init__(self, number, name, surname, location):
self.number = number
self.name = name
self.surname = surname
self.location = location
def isUnique(self, number):
test = Farmer.query.filter_by(number=number)
if test:
return True
else:
return False
class Unknown(db.Model):
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
number = db.Column(db.String(100))
name = db.Column(db.String(100))
seen = db.Column(db.Boolean)
time = db.Column(db.DateTime)
farmer_id = db.Column(db.Integer, db.ForeignKey('farmer.id'))
def __init__(self, content, number):
self.content = content
self.number = number
self.farmer_id = get_farmer(number)
self.name = get_farmer_name(number)
self.time = datetime.utcnow()
self.seen = False
class Question(db.Model):
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
response = db.Column(db.Text)
number = db.Column(db.String(100))
name = db.Column(db.String(100))
time = db.Column(db.DateTime)
farmer_id = db.Column(db.Integer, db.ForeignKey('farmer.id'))
def __init__(self, content, response, number):
self.content = content
self.response = response
self.number = number
self.farmer_id = get_farmer(number)
self.name = get_farmer_name(number)
self.time = datetime.utcnow()
class Disease(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(50))
category = db.Column(db.String(50))
symptoms = db.Column(db.Text)
remedy = db.Column(db.Text)
shona_remedy = db.Column(db.Text)
timestamp = db.Column(db.DateTime)
vet_name = db.Column(db.String(100))
def __init__(self, name, category, symptoms, remedy, shona_remedy, vet):
self.name = name
self.category = category
self.symptoms = symptoms
self.remedy = remedy
self.shona_remedy = shona_remedy
self.vet_name = vet
self.timestamp = datetime.utcnow()
def __repr__(self):
return '<{}>'.format(self.name)