from banjo.models import (
    Model, 
    StringField, 
    IntegerField, 
    FloatField,
    ForeignKey,
)
from django.db.models import DateTimeField

class Movie(Model):
    title = StringField()
    genre = StringField(blank = True, null = True)
    status = StringField(default = "Want to Watch")
    rating = IntegerField(blank = True, null = True)


    def to_dict(self):
        return {
            "id": self.id, 
            "title": self.title,
            "genre": self.genre,
            "status": self.status,
            "rating": self.rating,
        }