models.py

from tortoise.models import Model
from tortoise import fields


class Config(Model):
    id = fields.IntField(pk=True, unique=True)
    queue_upload_timeout = fields.FloatField()
    openai_api_key = fields.TextField()
    openai_base_prompt = fields.TextField()
    openai_question_regex = fields.TextField()
    recaptcha_sitekey = fields.TextField()
    twocaptcha_api_key = fields.TextField()


class OpenAIQuery(Model):
    id = fields.IntField(pk=True, unique=True)
    user = fields.ForeignKeyField('models.User', related_name='openai_queries')
    engine = fields.TextField()
    prompt = fields.TextField()
    temperature = fields.FloatField()
    max_tokens = fields.IntField()
    response = fields.TextField()


class QueueQuestion(Model):
    id = fields.IntField(pk=True, unique=True)
    title = fields.TextField()
    openai_query = fields.ForeignKeyField('models.OpenAIQuery', related_name='queue_questions')
    approved = fields.BooleanField()
    approved_user = fields.ForeignKeyField('models.User', null=True, related_name='queue_questions')
    upload_attempted = fields.BooleanField()


class User(Model):
    id = fields.IntField(pk=True, unique=True)
    name = fields.CharField(256, unique=True)
    password = fields.TextField()
    quora_user = fields.ForeignKeyField('models.QuoraUser', related_name='users')


class UserSession(Model):
    token = fields.CharField(256, pk=True, unique=True)
    user = fields.ForeignKeyField('models.User', related_name='session')
    expires = fields.DatetimeField()


class QuoraUser(Model):
    id = fields.IntField(pk=True, unique=True)
    email = fields.CharField(256, unique=True)
    password = fields.TextField()
    formkey = fields.CharField(256, unique=True)
    cookie = fields.CharField(256, unique=True)


class QuoraQuestion(Model):
    id = fields.CharField(256, pk=True, unique=True)
    qid = fields.IntField(unique=True)
    time_created = fields.DatetimeField()
    queue_question = fields.ForeignKeyField('models.QueueQuestion', null=True, related_name='quora_question')