diff options
Diffstat (limited to 'api/database/__init__.py')
| -rw-r--r-- | api/database/__init__.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/api/database/__init__.py b/api/database/__init__.py new file mode 100644 index 000000000..5b0bb05a2 --- /dev/null +++ b/api/database/__init__.py @@ -0,0 +1,23 @@ +############################################################################## +# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## +import logging + +from sqlalchemy import create_engine +from sqlalchemy.orm import scoped_session, sessionmaker +from sqlalchemy.ext.declarative import declarative_base + +logger = logging.getLogger(__name__) +logger.setLevel(logging.DEBUG) + +engine = create_engine('sqlite:////tmp/yardstick.db', convert_unicode=True) +db_session = scoped_session(sessionmaker(autocommit=False, + autoflush=False, + bind=engine)) +Base = declarative_base() +Base.query = db_session.query_property() |