#!/usr/bin/python
#-*- coding: utf-8 -*-

import datetime
from model1 import *

# Create/open database
db = create_database('mysql://mysql:@localhost/test')
conn = db.connect()

store = Store(db)

# Create Zoo objects
wap = Zoo(Name=u'Wild Animal Park',
          Founded=datetime.date(2000, 1, 1),
          Opens=datetime.time(8, 15, 59),
          LastEsc=datetime.datetime(2004, 7, 29, 5, 6, 7),
          Admission=4.95
    )
store.add(wap)
        
sdz = Zoo(Name = u'San  Diego  Zoo',
          Founded = datetime.date(1835, 9, 13),
          Opens = datetime.time(9, 0, 0),
          Admission = 0
    )
store.add(sdz)

print "--1- Zoo.ID, Zoo.Name before flush:"
print wap.ID, "\t", wap.Name
print sdz.ID, "\t", sdz.Name

store.flush()
store.commit()

print
print "--2- Zoo.ID, Zoo.Name after flush:"
for zoo in store.find(Zoo):
    print zoo.ID, "\t", zoo.Name.encode("latin1")

store.add(Animal(Species='Леопард', Lifespan=73.5, ZooID=sdz.ID))
store.add(Animal(Species='Тигр', Legs=1, Lifespan=.75, ZooID=sdz.ID))
store.add(Animal(Species='Медведь', ZooID=wap.ID))
store.add(Animal(Species='Лев', ZooID=wap.ID))
        
store.flush()
store.commit()

print
print "--3- Animal.ID, Animal.Species after flush:"
for animal in store.find(Animal):
    print animal.ID, "\t", animal.Species.encode("latin1")

print
print "--4- Zoo.Name, Animal.Species after search:"
for zoo in store.find((Zoo,Animal),"Zoo.ID=Animal.ZooID"):
    print zoo[0].Name.encode("latin1"), "\t", zoo[1].Species.encode("latin1")


store.close()