dataquant 1.1.6__py3-none-any.whl

dataquant/__init__.py

@@ -0,0 +1,150 @@
+# -*- coding: UTF-8 -*-
+
+import warnings
+import importlib
+import pkgutil
+import dataquant
+
+from dataquant.utils.client import init
+__version__ = '1.1.6'
+__all__ = []
+
+
+def _init():
+    for loader, module_name, is_pkg in pkgutil.walk_packages(__path__, "dataquant."):
+        if module_name.startswith("dataquant.apis") and not is_pkg:
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    globals()[api_name] = api
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.pof") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.ic") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.cd") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.sql") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.hk") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+        elif module_name.startswith("dataquant.intl") and not is_pkg:
+            namespace = module_name.split('.')[-2]
+            namespace_name = namespace
+            namespace = type(namespace, (object,), {})
+            namespace.__module__ = "dataquant"
+            setattr(dataquant, namespace_name, namespace)
+            dataquant.__all__.append(namespace_name)
+
+            try:
+                api_module = importlib.import_module(module_name)
+            except ImportError as ex:
+                warnings.warn("import module[{}] error, msg={}".format(module_name, ex))
+
+            _module = getattr(dataquant, namespace_name)
+            for api_name in api_module.__all__:
+                try:
+                    api = getattr(api_module, api_name)
+                    setattr(_module, api_name, api)
+                except AttributeError as ex:
+                    warnings.warn("load api[{}] error, msg={}".format(api_name, ex))
+
+_init()
+
+del _init

dataquant/apis/__init__.py

@@ -0,0 +1 @@
+# -*- coding: UTF-8 -*-

dataquant/apis/base/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+from dataquant.apis.base.api import init, get_data, get_qdata

dataquant/apis/base/api.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+
+__all__ = [
+    "init",
+    "environ",
+    "load_conf",
+    "get_data",
+    "get_qdata"
+]
+
+from dataquant.utils.config import read_config
+
+global QUOTE_FORMAT
+
+
+def load_conf():
+    global QUOTE_FORMAT
+    CONFIG = read_config()['system']
+    QUOTE_FORMAT = CONFIG.get('quote_format')
+
+
+load_conf()
+
+
+def init(**kwargs):
+    """
+    :param str username: 用户名，license模式下为'license'
+    :param str password: 用户密码，license模式下为下发的license
+    :param str protocol：传输协议，默认HTTP
+    :param str url: 数据服务的URL
+    :param int connect_timeout: 连接建立超时时间,默认5秒
+    :param int timeout: 数据传输超时时间，默认30秒
+    :param str compressor: 数据传输过程中的压缩算法，默认不使用
+    :param int max_pool_size: 连接池大小，默认10
+    :param int page_size: 数据请求分页大小，默认100000
+    :return:
+    """
+    from dataquant.utils.client import init as _init
+
+    _init(**kwargs)
+    return
+
+
+def environ():
+    """
+    获取环境配置信息
+    :return:
+    """
+    from dataquant.utils.client import environ as _environ
+    return _environ()
+
+
+def get_data(method, **kwargs):
+    """
+    共用获取数据接口
+    :param method:    接口名称
+    :param kwargs:    接口参数
+    """
+    from dataquant.utils.client import get_client
+    result = get_client().send(method, **kwargs)
+    if result['result_code'] == '0' or result['result_code'] == 0:
+        return result['data']
+    else:
+        raise Exception('异常代码[%s], 异常信息[%s]' % (result['result_code'], result['result_msg']))
+
+
+def get_qdata(method, **kwargs):
+    """
+    共用获取行情数据接口
+    :param method:    接口名称
+    :param kwargs:    接口参数
+    """
+    from dataquant.utils.client import get_client
+    global QUOTE_FORMAT
+    kwargs.setdefault('format', QUOTE_FORMAT)  # TabSeparatedWithNames Parquet Arrow
+    ret = get_client().send(method, **kwargs)
+    return ret

dataquant/apis/info/__init__.py

@@ -0,0 +1 @@
+# -*- coding: UTF-8 -*-