review code

Author: sfyc23

everyday_wechat/control/weather/sojson.py

@@ -4,11 +4,11 @@
 import json
 import os
 
+__all__ = ['get_sojson_weather', 'get_sojson_weather_tomorrow']
 
 with open(os.path.join(os.path.dirname(__file__), '_city_sojson.json'), 'r', encoding='utf-8') as f:
     CITY_CODE_DICT = json.loads(f.read())
 
-
 MSG_TOMORROW = '明日{city_name}天气\n{_date} {week}\n【明日天气】{_type}\n【明日气温】{low_temp} {high_temp}\n【明日风速】{speed}\n【出行提醒】{notice}'
 MSG_TODAY = '今日{city_name}天气\n{_date},{week}\n【今日天气】{_type}\n【今日气温】{low_temp} {high_temp}\n【今日风速】{speed}\n【出行提醒】{notice}'
 
@@ -80,7 +80,7 @@ def get_sojson_weather_tomorrow(city_name):
     """
     if not city_name:
         return None
-    city_code = city_dict.get(city_name, None)
+    city_code = CITY_CODE_DICT.get(city_name, None)
     if not city_code:
         print('没有此城市的消息...')
         return None

everyday_wechat/utils/config.py

@@ -7,6 +7,8 @@
 import copy as mycopy
 import yaml
 
+__all__ = ['set', 'get', 'copy', 'update', '_print']
+
 
 # def init():
 #     """
@@ -36,29 +38,35 @@ def get_yaml():
         print('你的 _config.yaml 文件配置出错...')
     return None
 
+
 opts = get_yaml()
 
+
 def set(key, value):
     """ 通过 key 设置某一项值 """
     opts[key] = value
 
+
 def get(key, default=None):
     """ 通过 key 获取值 """
     return opts.get(key, default)
 
+
 def copy():
     """ 复制配置 """
     return mycopy.deepcopy(opts)
 
+
 def update(new_opts):
     """ 全部替换配置 """
     opts.update(new_opts)
 
+
 def _print():
     print(opts)
 
+
 if __name__ == '__main__':
-    init()
     you = get('is_forced_switch')
     print(you)
     # print(copy())