#!/usr/bin/env python
import sys
from engine import engine
from seeders.base_seeder import BaseSeeder
import seeders.template
import seeders.game_engine
if __name__ == '__main__':
raw_seeders = BaseSeeder.__subclasses__()
seeders = {}
for raw_seeder in raw_seeders:
seeders[raw_seeder.__name__] = raw_seeder
session = engine.get_session()
if len(sys.argv) <= 1:
for seeder in seeders.values():
seeder(session).seed()
else:
if sys.argv[1] in seeders:
seeders[sys.argv[1]](session).seed()
else:
print('Invalid seeder given must be one of: %s' % ' '.join(seeders.keys()))
sys.exit(1)
session.commit()
def __init__(self, session):
BaseSeeder.__init__(self, session)
self.template_data = os.path.join(self.database_dir, 'templates')
def __init__(self, session):
BaseSeeder.__init__(self, session)
self.seed_data = os.path.join(self.database_dir, 'seed_data')
