def test0(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class E(db.Entity):
t2 = orm.Required('T2')
class T1(db.Entity):
t2 = orm.Required('T2')
t3 = orm.Optional('T3')
class T2(db.Entity):
t3 = orm.Required('T3')
t1 = orm.Optional('T1')
e = orm.Optional('E')
class T3(db.Entity):
t1 = orm.Required('T1')
t2 = orm.Optional('T2')
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test_2(self):
with open('migrations/0002.py', 'w') as f:
f.write('''
from pony.migrate import diagram_ops as op
from pony.orm import *
dependencies = ['0001_initial']
def forward(db):
db.execute('ALTER TABLE base MODIFY COLUMN s INTEGER')
db.execute('ALTER TABLE base DROP COLUMN classtype')
db.execute('RENAME TABLE base TO myentity')
def final_state(db):
class MyEntity(db.Entity):
int_f = Required(int)
s = Required(int)
operations = [
op.Custom(forward=forward, final_state=final_state),
]
''')
db = self.DB
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Test(db.Entity):
field = orm.Required(str)
class Parent(db.Entity):
_discriminator_ = 1
type_field = orm.Discriminator(int)
field = orm.Required(str)
orm.composite_index(type_field, field)
class Child(Parent):
_discriminator_ = 2
another_field = orm.Required(str)
orm.composite_key("type_field", another_field)
command.migrate(db, "make -v")
command.migrate(db, "apply -v")
with orm.db_session():
Parent(field="abcdef")
Child(field="qwerty", another_field="1234")
def test_0(self):
db_args, db_kwargs = self.db_params
self.__class__.DB = db = orm.Database(*db_args, **db_kwargs)
class MyEntity(db.Entity):
int_f = orm.Required(int)
command.migrate(db, 'make')
def test2(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
created = orm.Required(str)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test0(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
pass
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test0(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Activity(db.Entity):
descr = orm.Required(str)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
number = orm.Optional(int)
created = orm.Optional(int)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class MyEntity(db.Entity):
int_f = orm.Required(int)
str_f = orm.Required(str)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
self.__class__.DB = db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
label = orm.Optional(str)
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test0(self):
db_args, db_kwargs = self.db_params
self.__class__.DB = db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
created = orm.Required(datetime)
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test0(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Data(db.Entity):
label = orm.Required(str)
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test0(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Test(db.Entity):
field = orm.Required(str)
command.migrate(db, "make -v")
with open("migrations/0001_initial.py") as f:
print(f.read())
command.migrate(db, "apply -v")
def test0(self):
db_args, db_kwargs = self.db_params
db = self.__class__.db = orm.Database(*db_args, **db_kwargs)
class T1(db.Entity):
peers = orm.Set('T2')
class T2(db.Entity):
peers = orm.Set('T1')
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Activity(db.Entity):
s = orm.Optional(str)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
assert len(glob('migrations/*')) == 2
def test_1(self):
db_args, db_kwargs = self.db_params
self.__class__.DB = db = orm.Database(*db_args, **db_kwargs)
class MyEntity(db.Entity):
int_f = orm.Required(int)
s = orm.Required(str)
# db.generate_mapping()
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class T1(db.Entity):
peers = orm.Set('T2')
f = orm.Required(int, sql_default='3')
class T2(db.Entity):
peers = orm.Set('T1')
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test_2(self):
with open('migrations/0002.py', 'w') as f:
f.write('''
from pony.migrate import diagram_ops as op
dependencies = ['0001_initial']
operations = [
op.RenameAttr('MyEntity', 's', 't'),
op.RenameEntity('MyEntity', 'YourEntity'),
]
''')
command.migrate(self.DB, 'apply -v')
def test(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class MyEntity(db.Entity):
int_f = orm.Required(int)
str_f = orm.Required(str)
other = orm.Set('OtherEntity')
class OtherEntity(db.Entity):
fk = orm.PrimaryKey(MyEntity)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Activity(db.Entity):
descr = orm.Optional('Description')
class Description(db.Entity):
text = orm.Required(str)
activity = orm.Required('Activity')
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test_2(self):
with open('migrations/0002.py', 'w') as f:
f.write('''
from pony.migrate import diagram_ops as op
from pony.orm import *
dependencies = ['0001_initial']
operations = [
op.AddAttr('MyEntity', 's', Required(str, initial='initial')),
]
''')
command.migrate(self.DB, 'apply -v')
def test_1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class MyEntity(db.Entity):
int_f = orm.Required(int)
s = orm.Required(str)
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
[p] = glob.glob('./migrations/0002*')
with open(p) as f:
s = f.read()
self.assertTrue('op.AddEntity' in s)
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Activity(db.Entity):
descr = orm.Required(str)
periods = orm.Set('Time')
class Time(db.Entity):
amount = orm.Required(int)
spent_on = orm.Required('Activity')
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test_simple(self):
db_args, db_kwargs = self.db_params
with open('entities.py', 'w') as f:
f.write('''\
from pony.orm import *
db = Database(*{}, **{})
class MyEntity(db.Entity):
int_f = PrimaryKey(int)
str_f = Required(str)
'''.format(repr(db_args), repr(db_kwargs)))
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test2(self):
db_args, db_kwargs = self.db_params
self.__class__.DB = db = orm.Database(*db_args, **db_kwargs)
class News(db.Entity):
content = orm.Required(str)
source = orm.Required('Source')
class Source(db.Entity):
name = orm.Required(str)
info = orm.Optional('News')
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test2(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class Activity(db.Entity):
descr = orm.Required(str)
class Time(db.Entity):
amount = orm.Required(int)
spent_on = orm.Required(str)
db.generate_mapping(create_tables=False, check_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
db = orm.Database(*db_args, **db_kwargs)
class U(db.Entity):
peers = orm.Set('T2')
f = orm.Required(int)
class T2(db.Entity):
peers = orm.Set('U')
db.generate_mapping(check_tables=False, create_tables=False)
command.migrate(db, 'make -v')
command.migrate(db, 'apply -v')
def test1(self):
db_args, db_kwargs = self.db_params
p = os.path.join('migrations', '0002.py')
with open(p, 'w') as f:
f.write('''\
from pony import orm
from pony.migrate import diagram_ops as op
dependencies = ['0001_initial']
operations = [
op.AddAttr('T1', 'f', orm.Optional(int, initial=True)),
]
'''.format(db_args, db_kwargs))
command.migrate(self.db, 'apply -v')
def test_1(self):
db_args, db_kwargs = self.db_params
with open('migrations/0002.py', 'w') as f:
f.write('''
from pony.migrate import diagram_ops as ops
from pony.orm import *
dependencies = ['0001_initial']
operations = [
ops.AddAttr('MyEntity', 's', Required(str, initial='S'))
]
''')
db = self.DB
command.migrate(db, 'apply -v')