models.py

# todo: warn on no default, or something
# todo: get default collection name for picklestorage, mongostorage constructors
# todo: requirements.txt
# todo: distutils

import pprint

from modularodm import StoredObject
from modularodm.fields.StringField import StringField
from modularodm.fields.IntegerField import IntegerField
from modularodm.fields.FloatField import FloatField
from modularodm.fields.BooleanField import BooleanField
from modularodm.fields.DateTimeField import DateTimeField
from modularodm.fields.ForeignField import ForeignField
from modularodm.storage.PickleStorage import PickleStorage
from modularodm.storage.MongoStorage import MongoStorage
from modularodm.validators import *
from modularodm.query.querydialect import DefaultQueryDialect as Q

pp = pprint.PrettyPrinter(indent=4)

import random
import logging
debug = True

logging.basicConfig(level=logging.DEBUG)
if debug:
    import os
    try:os.remove('db_team.pkl')
    except:pass
    try:os.remove('db_manager.pkl')
    except:pass
    try:os.remove('db_player.pkl')
    except:pass

class Team(StoredObject):
    name = StringField(primary=True)
    owner = ForeignField('Manager', backref='owned')
    wins = IntegerField(list=True)
    playoffs = BooleanField(default=None, list=True)
    schedule = StringField(list=True)
    players = ForeignField('Player', list=True, backref='plays_for')

class Manager(StoredObject):
    name = StringField(primary=True)
    players_managed = ForeignField('Player', list=True, backref='managed_by')

class Player(StoredObject):
    name = StringField(primary=True)
    number = IntegerField()
    rating = FloatField(default=0.0)
    injured = BooleanField(default=False)

Team.set_storage(PickleStorage('team'))
Manager.set_storage(PickleStorage('manager'))
Player.set_storage(PickleStorage('player'))

if debug:
    d = Player(name="Griffin", number=10, rating=85.0, injured=True)
    d.save()

    e = Player(name="Morris", number=46, rating=80.2, injured=False)
    e.save()

    f = Player(name="Moss", number=89, rating=82.7, injured=False)
    f.save()

    g = Player(name="Tannehill", number=17, rating=75.0, injured=False)
    g.save()

    h = Player(name="Wallace", number=11, rating=80.0, injured=False)
    h.save()

    i = Player(name="Wake", number=91, rating=84.4, injured=False)
    i.save()

    j = Player(name="Cutler", number=6, rating=70.2, injured=False)
    j.save()

    k = Player(name="Marshall", number=15, rating=77.0, injured=False)
    k.save()

    l = Player(name="Collins", number=93, rating=72.4, injured=False)
    l.save()

    a = Manager(name="finsfan", players_managed=[d, g, h, i])
    a.save()

    b = Manager(name="skinsfan", players_managed=[d, e, f])
    b.save()

    c = Manager(name="bearsfan", players_managed=[j, k, l])
    c.save()

    m = Team(name="FinsRock", owner=a, wins=[3,1,2], playoffs=[True, False, True], schedule=["Home", "Away", "Away"], players=[d, g, h, i])
    m.save()

    n = Team(name="SkinsRock", owner=b, wins=[2,0,1], playoffs=[True, False, False], schedule=["Home", "Away", "Home"], players=[d, e, f])
    n.save()

    o = Team(name="BearsRock", owner=c, wins=[3,1,4], playoffs=[False, False, False], schedule=["Away", "Away", "Home"], players=[j, k, l])
    o.save()