class Keyword(db.Model):
__tablename__ = 'keywords'
id = db.Column(db.Integer, primary_key=True)
word = db.Column(db.String(20), nullable=False, unique=True, index=True)
users = db.relationship('User',
secondary=choices,
back_populates='keywords',
lazy='dynamic')
class EmailActive(db.Model):
__tablename__ = 'email_active'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer,
db.ForeignKey('users.id'),
unique=True,
nullable=False)
active_code = db.Column(db.String(24), nullable=False)
is_active = db.Column(db.Boolean, nullable=False)
def set_active(self, status):
self.is_active = status
db.session.commit()
class User(UserMixin, db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(30), nullable=False)
password = db.Column(db.String(30), nullable=False)
email = db.Column(db.String(40), nullable=False, unique=True, index=True)
keywords = db.relationship('Keyword',
secondary=choices,
back_populates='users',
lazy='dynamic')
email_active = db.relationship('EmailActive',
uselist=False,
backref='user')
@staticmethod
def save_user(user_row):
db.session.add(user_row)
db.session.commit()
return user_row.id
@staticmethod
def delete_user(user_id):
email_active = EmailActive.query.filter_by(user_id=user_id).first()
if email_active is not None:
db.session.delete(email_active)
db.session.commit()
user = User.query.filter_by(id=user_id).first()
if user is not None:
db.session.delete(user)
db.session.commit()
return True
else:
return False
@staticmethod
def save_email_active(email_active_row):
db.session.add(email_active_row)
db.session.commit()
return email_active_row.id
def delete_keyword_by_word(self, word_to_delete):
for keyword_row in self.keywords:
if keyword_row.word == word_to_delete:
self.keywords.remove(keyword_row)
db.session.commit()
return True
return False
def save_keyword(self, word_to_save):
keyword_in_table = Keyword.query.filter_by(word=word_to_save).first()
if word_to_save in [keyword.word for keyword in self.keywords]:
return False
if keyword_in_table is None:
keyword_in_table = Keyword(word=word_to_save)
db.session.add(keyword_in_table)
self.keywords.append(keyword_in_table)
db.session.commit()
return True
@staticmethod
def get_all_email():
query_result = User.query.with_entities(User.email).all()
email_list = [item[0] for item in query_result]
return email_list
class TuanweiInfo(db.Model):
__tablename__ = 'tuanwei_info'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
url = db.Column(db.String(255), nullable=False)
class_ = db.Column('class', db.Integer, nullable=False)