class FeedbackContent(db.EmbeddedDocument):
type = db.StringField(required=True)
ageFeedback = db.StringField(required=True)
ageCorrectness = db.BooleanField(required=True)
genderFeedback = db.StringField(required=True)
genderCorrectness = db.BooleanField(required=True)
emotionFeedback = db.StringField(required=True)
emotionCorrectness = db.BooleanField(required=True)
class FATask(db.Document):
wav_url = db.StringField(required=False, dafault='')
wav_tmp_path = db.StringField(required=False, dafault='/tmp/tmppath')
request_srt_content = db.StringField(required=False, dafault='')
result_srt_content = db.StringField(required=False, dafault='')
steps = db.EmbeddedDocumentListField(TaskStep)
timestamp = db.LongField(required=False, default=time.time)
segment = db.BooleanField(required=False, dafault=False)
meta = {
'collection': 'fa_task',
'index_background': True,
'indexes': ['timestamp'],
'strict': False
}
def add_step(self, step_obj):
self.steps.append(step_obj)
return self.save()
def to_dict(self):
json_obj = dict(
self.to_mongo(fields=[
'wav_url', 'request_srt_content', 'result_srt_content',
'steps', 'timestamp', 'segment'
]))
json_obj.pop('_id', '')
json_obj['steps'] = []
for step in self.steps:
json_obj['steps'].append(step.to_dict())
json_obj['task_id'] = str(self.pk)
return json_obj
class User(UserMixin, db.Document):
meta = {"collection": "users"}
id = db.StringField(default=lambda: str(uuid.uuid4()), primary_key=True)
active = db.BooleanField(default=True)
# User authentication information
email = db.EmailField()
password = db.StringField()
# User information
username = db.StringField(default="")
# User image
image = db.StringField(default="default.jpg")
# Relationships
posts = db.ListField(db.StringField(), default=[])
def get_reset_token(self, expires_sec=1800):
s = Serializer(app.config["SECRET_KEY"], expires_sec)
return s.dumps({"user_id": self.id}).decode("utf-8")
@staticmethod
def verify_reset_token(token):
s = Serializer(app.config["SECRET_KEY"])
try:
user_id = s.loads(token)["user_id"]
except:
return None
return User.objects.get(id=user_id)
class Task(db.EmbeddedDocument):
title = db.StringField(verbose_name="Title",
max_length=255,
unique=True,
required=True)
details = db.StringField(verbose_name="Details")
setdate = db.DateTimeField(default=datetime.datetime.now)
lastdate = db.DateTimeField(verbose_name="Last Date")
completed = db.BooleanField(default=False)
url = db.StringField(max_length=255, unique=True)
class Newsletter(db.Document):
option = db.BooleanField(required=True) # Sim ou Nao
tipo = db.StringField(required=True) # Shows Locais, Meus Shows, etc..
user = db.ReferenceField(User, required=True, dbref=False)
def __eq__(self, other):
return self.user == other.user and self.option == other.option and self.tipo == other.tipo
def __unicode__(self):
return u"%s - %s - %s" % (self.user, self.tipo,
u"Sim" if self.option else u"Não")
class User(db.Document):
phone_number = db.StringField(required=True, unique=True)
first_name = db.StringField(required=True)
last_name = db.StringField(required=True)
email = db.StringField(required=True, unique=True)
is_active = db.BooleanField(default=False)
password = db.StringField(required=True)
api_token = db.StringField()
user_role = db.StringField(default="store_admin")
reg_date = db.DateTimeField(default=datetime.datetime.utcnow)
def __str__(self):
return "<User: {}>".format(self.first_name)