50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
|
from peewee import *
|
||
|
|
from playhouse.sqlite_ext import SqliteExtDatabase
|
||
|
|
import datetime
|
||
|
|
import json
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
def coloursToJson(colours):
|
||
|
|
colours2 = [(list(colour[0]), colour[1]) for colour in colours]
|
||
|
|
return json.dumps(colours2)
|
||
|
|
|
||
|
|
def jsonToColours(string):
|
||
|
|
data = json.loads(string)
|
||
|
|
return [(np.array(d[0]), d[1]) for d in data]
|
||
|
|
|
||
|
|
db = SqliteExtDatabase('images.db')
|
||
|
|
|
||
|
|
class ColoursField(TextField):
|
||
|
|
# db_field = 'colour'
|
||
|
|
|
||
|
|
def db_value(self, value):
|
||
|
|
return coloursToJson(value)
|
||
|
|
|
||
|
|
def python_value(self, value):
|
||
|
|
return jsonToColours(value) # convert str to UUID
|
||
|
|
|
||
|
|
|
||
|
|
class BaseModel(Model):
|
||
|
|
class Meta:
|
||
|
|
database = db
|
||
|
|
|
||
|
|
class Emotion(BaseModel):
|
||
|
|
name = CharField(unique=True)
|
||
|
|
|
||
|
|
class Group(BaseModel):
|
||
|
|
name = CharField(unique=True)
|
||
|
|
|
||
|
|
class Artwork(BaseModel):
|
||
|
|
author = CharField()
|
||
|
|
age = SmallIntegerField(index=True)
|
||
|
|
gender = FixedCharField(max_length=1) # we should not really use this one
|
||
|
|
group = ForeignKeyField(Group, related_name='artworks', index=True)
|
||
|
|
emotion = ForeignKeyField(Emotion, related_name='artworks', index=True)
|
||
|
|
created_date = DateTimeField(default=datetime.datetime.now)
|
||
|
|
filename = CharField()
|
||
|
|
colours = ColoursField() # serialised colours + percentages: [([r,g,b], percentage), ...]
|
||
|
|
|
||
|
|
def getAges():
|
||
|
|
r = Artwork.select(fn.Distinct(Artwork.age)).dicts()
|
||
|
|
return [a['age'] for a in r]
|