🔨 use black to format code

Author: shidenggui

cli.py

@@ -7,36 +7,46 @@
 
 import easytrader
 
-ACCOUNT_OBJECT_FILE = 'account.session'
+ACCOUNT_OBJECT_FILE = "account.session"
 
 
 @click.command()
-@click.option('--use', help='指定券商 [ht, yjb, yh]')
-@click.option('--prepare', type=click.Path(exists=True), help='指定登录账户文件路径')
-@click.option('--get', help='调用 easytrader 中对应的变量')
-@click.option('--do', help='调用 easytrader 中对应的函数名')
-@click.option('--debug', default=False, help='是否输出 easytrader 的 debug 日志')
-@click.argument('params', nargs=-1)
+@click.option("--use", help="指定券商 [ht, yjb, yh]")
+@click.option("--prepare", type=click.Path(exists=True), help="指定登录账户文件路径")
+@click.option("--get", help="调用 easytrader 中对应的变量")
+@click.option("--do", help="调用 easytrader 中对应的函数名")
+@click.option("--debug", default=False, help="是否输出 easytrader 的 debug 日志")
+@click.argument("params", nargs=-1)
 def main(prepare, use, do, get, params, debug):
     if get is not None:
         do = get
-    if prepare is not None and use in ['ht_client', 'yjb', 'yh_client','yh','ht', 'gf', 'xq']:
+    if prepare is not None and use in [
+        "ht_client",
+        "yjb",
+        "yh_client",
+        "yh",
+        "ht",
+        "gf",
+        "xq",
+    ]:
         user = easytrader.use(use, debug)
         user.prepare(prepare)
-        with open(ACCOUNT_OBJECT_FILE, 'wb') as f:
+        with open(ACCOUNT_OBJECT_FILE, "wb") as f:
             dill.dump(user, f)
     if do is not None:
-        with open(ACCOUNT_OBJECT_FILE, 'rb') as f:
+        with open(ACCOUNT_OBJECT_FILE, "rb") as f:
             user = dill.load(f)
 
         if get is not None:
             result = getattr(user, do)
         else:
             result = getattr(user, do)(*params)
 
-        json_result = json.dumps(result, indent=4, ensure_ascii=False, sort_keys=True)
+        json_result = json.dumps(
+            result, indent=4, ensure_ascii=False, sort_keys=True
+        )
         click.echo(json_result)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

easytrader/__init__.py

@@ -5,5 +5,5 @@
 from .ricequant_follower import RiceQuantFollower
 from . import exceptions
 
-__version__ = '0.14.2'
-__author__ = 'shidenggui'
+__version__ = "0.14.2"
+__author__ = "shidenggui"

easytrader/api.py

@@ -9,7 +9,7 @@
 from .xqtrader import XueQiuTrader
 
 if six.PY2:
-    raise TypeError('不支持 Python2，请升级 Python3 ')
+    raise TypeError("不支持 Python2，请升级 Python3 ")
 
 
 def use(broker, debug=True, **kwargs):
@@ -27,19 +27,23 @@ def use(broker, debug=True, **kwargs):
     """
     if not debug:
         log.setLevel(logging.INFO)
-    elif broker.lower() in ['xq', '雪球']:
+    elif broker.lower() in ["xq", "雪球"]:
         return XueQiuTrader(**kwargs)
-    elif broker.lower() in ['yh_client', '银河客户端']:
+    elif broker.lower() in ["yh_client", "银河客户端"]:
         from .yh_clienttrader import YHClientTrader
+
         return YHClientTrader()
-    elif broker.lower() in ['ht_client', '华泰客户端']:
+    elif broker.lower() in ["ht_client", "华泰客户端"]:
         from .ht_clienttrader import HTClientTrader
+
         return HTClientTrader()
-    elif broker.lower() in ['gj_client', '国金客户端']:
+    elif broker.lower() in ["gj_client", "国金客户端"]:
         from .gj_clienttrader import GJClientTrader
+
         return GJClientTrader()
-    elif broker.lower() in ['ths', '同花顺客户端']:
+    elif broker.lower() in ["ths", "同花顺客户端"]:
         from .clienttrader import ClientTrader
+
         return ClientTrader()
 
 
@@ -61,9 +65,9 @@ def follower(platform, **kwargs):
         >>> jq.login(user='username', password='password')
         >>> jq.follow(users=user, strategies=['strategies_link'])
     """
-    if platform.lower() in ['rq', 'ricequant', '米筐']:
+    if platform.lower() in ["rq", "ricequant", "米筐"]:
         return RiceQuantFollower()
-    if platform.lower() in ['jq', 'joinquant', '聚宽']:
+    if platform.lower() in ["jq", "joinquant", "聚宽"]:
         return JoinQuantFollower()
-    if platform.lower() in ['xq', 'xueqiu', '雪球']:
+    if platform.lower() in ["xq", "xueqiu", "雪球"]:
         return XueQiuFollower(**kwargs)