yunorunner/models.py

import peewee

db = peewee.SqliteDatabase('db.sqlite')


class Repo(peewee.Model):
    name = peewee.CharField()  # TODO make this uniq/index
    url = peewee.CharField()
    revision = peewee.CharField(null=True)

    state = peewee.CharField(choices=(
        ('working', 'Working'),
        ('other_than_working', 'Other than working'),
    ), default="other_than_working")

    random_job_day = peewee.IntegerField(null=True)

    class Meta:
        database = db


class Job(peewee.Model):
    name = peewee.CharField()
    url_or_path = peewee.CharField()

    state = peewee.CharField(choices=(
        ('scheduled', 'Scheduled'),
        ('runnning', 'Running'),
        ('done', 'Done'),
        ('failure', 'Failure'),
        ('error', 'Error'),
        ('canceled', 'Canceled'),
    ), default="scheduled")

    log = peewee.TextField(default="")

    created_time = peewee.DateTimeField(constraints=[peewee.SQL("DEFAULT (datetime('now'))")])
    started_time = peewee.DateTimeField(null=True)
    end_time = peewee.DateTimeField(null=True)

    class Meta:
        database = db


class Worker(peewee.Model):
    state = peewee.CharField(choices=(
        ('available', 'Available'),
        ('busy', 'Busy'),
    ))

    class Meta:
        database = db


# peewee is a bit stupid and will crash if the table already exists
for i in [Repo, Job, Worker]:
    try:
        i.create_table()
    except Exception:
        pass