Пример #1
0
class TestConfigManager(unittest.TestCase):

    def setUp(self):
        self.src = TestSource('test', x=5)
        self.mgr = ConfigManager('prod', self.src)

    def test_get_version(self):
        vobj = self.mgr['2.0']
        assert isinstance(vobj, ConfigVersion)
        assert vobj.version == 4

        vobj2 = self.mgr.get_version(pydt.datetime(2012, 2, 1))
        assert vobj.tag == '2.0'
        assert vobj is vobj2

        vobj = self.mgr.get_version(2)
        assert vobj.tag is None
        assert vobj.version == 2

    def test_get_config(self):
        name = 'us_largecap'
        date = pydt.datetime(2012, 1, 31)
        vdate = pydt.datetime(2012, 1, 29)
        item = self.mgr.get_config(name, 'quantmodel', date) # 2012-1-31
        assert isinstance(item, ConfigItem)
        assert item.universe == 'SPX'
        assert item.key == 4

        item_v0 = self.mgr.get_config(name, 'quantmodel', date, '1.0')
        assert isinstance(item_v0, ConfigItem)
        assert item_v0.key == 0

        item_v8 = self.mgr.get_config(name, 'quantmodel', date, 8)
        assert isinstance(item_v8, ConfigItem)
        assert item_v8.key == 8

        item_v4 = self.mgr.get_config(name, 'quantmodel', date, vdate)
        assert isinstance(item_v4, ConfigItem)
        assert item_v4.key == 4
Пример #2
0
def _init_database():
    """
    初始化 SQLAlchemy for MySQL
    """
    config = ConfigManager.get_config()

    # 初始化数据库连接
    engine_1 = engine_from_config(config.DB_1_CONFIG, prefix='')

    # 创建DBSession类型
    session_1_factory = sessionmaker(bind=engine_1, autoflush=False)

    # 托管到 scoped_session
    session_read_class = scoped_session(session_1_factory)

    return session_read_class
Пример #3
0
def get_conn_basicos():
    conf = ConfigManager.get_config('db', "BASICOS")
    conn = pymysql.connect(host=conf['host'],
                           user=conf['user'], passwd=conf['password'], db=conf['database'],
                           charset='utf8')
    return conn
Пример #4
0
def get_conn_basicos():
    conf = ConfigManager.get_config('db', "BASICOS")
    conn = pymysql.connect(host=conf['host'],
                           user=conf['user'], passwd=conf['password'], db=conf['database'],
                           charset='utf8')
    return conn


def query(s, array=None):
    try:
        dbConn = get_conn_basicos()
        cursor = dbConn.cursor()
        cursor.execute(s, array)
        res = []
        for x in cursor.fetchall():
            res.append([xx for xx in x])
        dbConn.commit()
        cursor.close()
        return res
    except Exception as e:
        return [[e]]


medicine_conn = DbConn(ConfigManager.get_config("db", "MEDICINE"))
comparator_conn = DbConn(ConfigManager.get_config("db", "COMPARATOR"))
# basicos_conn = DbConn(ConfigManager.get_config('db', "BASICOS"))

if __name__ == '__main__':
    print(query('select 7;'))