def init_database(self):
self.local_database = '/tmp/orator_test_database.db'
if os.path.exists(self.local_database):
os.remove(self.local_database)
self.manager = DatabaseManager({
'default': 'sqlite',
'sqlite': {
'driver': 'sqlite',
'database': self.local_database
}
})
with self.manager.transaction():
try:
self.manager.statement(
'CREATE TABLE `users` ('
'id INTEGER PRIMARY KEY NOT NULL, '
'name CHAR(50) NOT NULL, '
'created_at DATETIME DEFAULT CURRENT_TIMESTAMP, '
'updated_at DATETIME DEFAULT CURRENT_TIMESTAMP'
')'
)
except Exception:
pass
Model.set_connection_resolver(self.manager)
self.manager.disconnect()
def init_database(self):
self.local_database = "/tmp/orator_test_database.db"
if os.path.exists(self.local_database):
os.remove(self.local_database)
self.manager = DatabaseManager({
"default": "sqlite",
"sqlite": {
"driver": "sqlite",
"database": self.local_database
},
})
with self.manager.transaction():
try:
self.manager.statement(
"CREATE TABLE `users` ("
"id INTEGER PRIMARY KEY NOT NULL, "
"name CHAR(50) NOT NULL, "
"created_at DATETIME DEFAULT CURRENT_TIMESTAMP, "
"updated_at DATETIME DEFAULT CURRENT_TIMESTAMP"
")")
except Exception:
pass
Model.set_connection_resolver(self.manager)
self.manager.disconnect()
class OratorTestCase(TestCase):
def tearDown(self):
if hasattr(self, "local_database"):
os.remove(self.local_database)
flexmock_teardown()
def init_database(self):
self.local_database = "/tmp/orator_test_database.db"
if os.path.exists(self.local_database):
os.remove(self.local_database)
self.manager = DatabaseManager({
"default": "sqlite",
"sqlite": {
"driver": "sqlite",
"database": self.local_database
},
})
with self.manager.transaction():
try:
self.manager.statement(
"CREATE TABLE `users` ("
"id INTEGER PRIMARY KEY NOT NULL, "
"name CHAR(50) NOT NULL, "
"created_at DATETIME DEFAULT CURRENT_TIMESTAMP, "
"updated_at DATETIME DEFAULT CURRENT_TIMESTAMP"
")")
except Exception:
pass
Model.set_connection_resolver(self.manager)
self.manager.disconnect()
def assertRegex(self, *args, **kwargs):
if PY2:
return self.assertRegexpMatches(*args, **kwargs)
else:
return super(OratorTestCase, self).assertRegex(*args, **kwargs)
def assertNotRegex(self, *args, **kwargs):
if PY2:
return self.assertNotRegexpMatches(*args, **kwargs)
else:
return super(OratorTestCase, self).assertNotRegex(*args, **kwargs)
class OratorTestCase(TestCase):
def tearDown(self):
if hasattr(self, 'local_database'):
os.remove(self.local_database)
def init_database(self):
self.local_database = '/tmp/orator_test_database.db'
if os.path.exists(self.local_database):
os.remove(self.local_database)
self.manager = DatabaseManager({
'default': 'sqlite',
'sqlite': {
'driver': 'sqlite',
'database': self.local_database
}
})
with self.manager.transaction():
try:
self.manager.statement(
'CREATE TABLE `users` ('
'id INTEGER PRIMARY KEY NOT NULL, '
'name CHAR(50) NOT NULL, '
'created_at DATETIME DEFAULT CURRENT_TIMESTAMP, '
'updated_at DATETIME DEFAULT CURRENT_TIMESTAMP'
')'
)
except Exception:
pass
Model.set_connection_resolver(self.manager)
self.manager.disconnect()
def assertRegex(self, *args, **kwargs):
if PY2:
return self.assertRegexpMatches(*args, **kwargs)
else:
return super(OratorTestCase, self).assertRegex(*args, **kwargs)
def assertNotRegex(self, *args, **kwargs):
if PY2:
return self.assertNotRegexpMatches(*args, **kwargs)
else:
return super(OratorTestCase, self).assertNotRegex(*args, **kwargs)
def init_database(self):
self.local_database = '/tmp/orator_test_database.db'
if os.path.exists(self.local_database):
os.remove(self.local_database)
self.manager = DatabaseManager({
'default': 'sqlite',
'sqlite': {
'driver': 'sqlite',
'database': self.local_database
}
})
with self.manager.transaction():
try:
self.manager.statement(
'CREATE TABLE `users` ('
'id INTEGER PRIMARY KEY NOT NULL, '
'name CHAR(50) NOT NULL, '
'created_at DATETIME DEFAULT CURRENT_TIMESTAMP, '
'updated_at DATETIME DEFAULT CURRENT_TIMESTAMP'
')'
)
except Exception:
pass
Model.set_connection_resolver(self.manager)
self.manager.disconnect()
def _get_real_manager(self):
manager = DatabaseManager({
'default': 'sqlite',
'sqlite': {
'driver': 'sqlite',
'database': ':memory:'
}
})
return manager
def _get_real_manager(self):
manager = DatabaseManager({
"default": "sqlite",
"sqlite": {
"driver": "sqlite",
"database": ":memory:"
},
})
return manager
def __init__(self):
"""
Initialize the Orator Migrator Class. Check if the migration table has been created. If not, Create the
Repository
"""
check, config = self.get_config()
if not check:
print("Error Occurred")
else:
self.manager = DatabaseManager(config=config)
self.path = os.path.abspath('.') + "/database/migrations/"
self.repository = DatabaseMigrationRepository(resolver=self.manager, table='migrations')
if not self.repository.repository_exists():
self.repository.create_repository()
super().__init__(self.repository, self.manager)
import environs
from orator import Model, SoftDeletes, mutator
from orator.orm import (has_one, has_many, belongs_to, accessor,
belongs_to_many, scope, has_many_through,
morphed_by_many, morph_to_many)
from orator.database_manager import DatabaseManager
env = environs.Env()
log = logging.getLogger(__name__)
db = DatabaseManager({
'postgres': {
'driver': 'postgres',
'host': env('POSTGRES_HOST'),
'port': env('POSTGRES_PORT'),
'database': env('POSTGRES_DB'),
'user': env('POSTGRES_USER'),
'password': env('POSTGRES_PASSWORD'),
'prefix': '',
'log_queries': True
}
})
Model.set_connection_resolver(db)
class Trainers(Model):
__fillable__ = ['name', 'gender', 'rival_name']
__hidden__ = ['created_at', 'updated_at']
__appends__ = []
_with = ['pokemon']
def get_db():
config = yaml.load(open("orator.yaml"))
return DatabaseManager(config["databases"])
def load_db():
config = yaml.load(open("orator.yaml"))
db = DatabaseManager(config["databases"])
Model.set_connection_resolver(db)
return db
