TutorialDivingIn05: model.py

from elixir import *

metadata.bind = 'sqlite:///movies.sqlite'
metadata.bind.echo = True

class Movie(Entity):
    title = Field(Unicode(30), primary_key=True)
    year = Field(Integer, primary_key=True)
    description = Field(Unicode)
    releasedate = Field(DateTime)
    
    director = ManyToOne('Director')
    genres = ManyToMany('Genre')
    actors = ManyToMany('Actor', tablename='movie_casting')

    def __repr__(self):
        return '<Movie "%s" (%d)>' % (self.title, self.year)

class Person(Entity):
    using_options(inheritance='multi')

    name = Field(Unicode(60))

    def __repr__(self):
        return '<Person "%s">' % self.name

class Actor(Person):
    using_options(inheritance='multi')

    movies = ManyToMany('Movie', tablename='movie_casting')

    def __repr__(self):
        return '<Actor "%s">' % self.name

class Director(Person):
    using_options(inheritance='multi')

    movies = OneToMany('Movie')

    def __repr__(self):
        return '<Director "%s">' % self.name

class Genre(Entity):
    name = Field(Unicode(15), primary_key=True)

    movies = ManyToMany('Movie')
    
    def __repr__(self):
        return '<Genre "%s">' % self.name