settings.py 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220
  1. from dotenv import load_dotenv, find_dotenv
  2. from pathlib import Path # Python 3.6+ only
  3. import configparser, json, pytz
  4. TIME_ZONE = pytz.timezone("Asia/Shanghai")
  5. from numpy import true_divide
  6. from databases import (
  7. create_all_database,
  8. DeviceConfig,
  9. SysConfigs,
  10. CRUD,
  11. batch_insert_sys_configs,
  12. SqlQuery,
  13. batch_insert_device_configs,
  14. )
  15. # 初始化数据表
  16. create_all_database()
  17. session = SqlQuery()
  18. device_config_crud = CRUD(DeviceConfig)
  19. all_devices = device_config_crud.read_all(session)
  20. if len(all_devices) == 0:
  21. # 如果配置表中一条数据都没有,就将初始化数据全部插入到数据表中
  22. actions = json.load(open("action.json", encoding="utf-8"))
  23. batch_insert_device_configs(session, actions)
  24. sys_config_crud = CRUD(SysConfigs)
  25. all_sys_configs = sys_config_crud.read_all(session)
  26. if len(all_sys_configs) == 0:
  27. # 如果配置表中一条数据都没有,就将初始化数据全部插入到数据表中
  28. sys_config_json = json.load(open("sys_configs.json", encoding="utf-8"))
  29. batch_insert_sys_configs(session, sys_config_json)
  30. # 初始化数据表---结束
  31. def get_config_by_items(config_dict):
  32. __config_dict = {}
  33. for i, k in config_dict:
  34. __config_dict[i] = k
  35. return __config_dict
  36. def getSysConfigs(key,item,default=None):
  37. session = SqlQuery()
  38. crud = CRUD(SysConfigs)
  39. one_item = crud.read(session, conditions={"key": key})
  40. config = json.loads(one_item.value)
  41. return config.get(item, default)
  42. def get_dict_value(_dict, key, default=None):
  43. if key in _dict:
  44. return _dict[key]
  45. else:
  46. return default
  47. MACHINE_LEVEL = getSysConfigs("other_configs", "device_speed", "二档")
  48. IS_TEST = False
  49. IS_MCU = True
  50. IS_LIN_SHI_TEST = False
  51. PhotographSeconds = float(
  52. getSysConfigs("take_photo_configs", "camera_delay", "0.5")
  53. ) # 拍照停留时间
  54. def moveSpeed(level: str = None):
  55. config = {
  56. "一档": {
  57. "camera_high_motor": {
  58. "max_speed": 10000,
  59. "up_speed": 800,
  60. "down_speed": 700,
  61. },
  62. "turntable_steering": {
  63. "max_speed": 6000,
  64. "up_speed": 500,
  65. "down_speed": 400,
  66. },
  67. },
  68. "二档": {
  69. "camera_high_motor": {
  70. "max_speed": 7000,
  71. "up_speed": 600,
  72. "down_speed": 500,
  73. },
  74. "turntable_steering": {
  75. "max_speed": 4500,
  76. "up_speed": 350,
  77. "down_speed": 300,
  78. },
  79. },
  80. "三档": {
  81. "camera_high_motor": {
  82. "max_speed": 3500,
  83. "up_speed": 400,
  84. "down_speed": 300,
  85. },
  86. "turntable_steering": {
  87. "max_speed": 3000,
  88. "up_speed": 200,
  89. "down_speed": 200,
  90. },
  91. },
  92. }
  93. if level is None:
  94. return config[MACHINE_LEVEL]
  95. else:
  96. return config[level]
  97. config = configparser.ConfigParser()
  98. config_name = "config.ini"
  99. config.read(config_name, encoding="utf-8")
  100. # 应用名称
  101. APP_NAME = config.get("app", "app_name")
  102. # 应用版本号
  103. APP_VERSION = config.get("app", "version")
  104. # 是否开启调试模式
  105. IS_DEBUG = config.get("app", "debug")
  106. # 应用端口号
  107. PORT = config.get("app", "port")
  108. # 应用线程数
  109. APP_WORKS = config.get("app", "works")
  110. # 应用host地址
  111. APP_HOST = config.get("app", "host")
  112. # 应用服务启动名称
  113. APP_RUN = config.get("app", "app_run")
  114. # 日志名称
  115. LOG_FILE_NAME = config.get("log", "log_file_name")
  116. # 最大字节数
  117. MAX_BYTES = config.get("log", "max_bytes")
  118. print("Max bytes is", MAX_BYTES)
  119. # 备份数量
  120. BACKUP_COUNTS = config.get("log", "backup_counts")
  121. # 远程服务器地址
  122. HLM_HOST = config.get("log", "hlm_host")
  123. PROJECT = config.get("app", "project")
  124. # ----------------------------------
  125. mcu_config_dict = config.items("mcu_config")
  126. _mcu_config_dict = {}
  127. for i, k in mcu_config_dict:
  128. _mcu_config_dict[i] = int(k)
  129. # print(_mcu_config_dict)
  130. _config_mcu_config = get_config_by_items(config.items("mcu_config"))
  131. LEFT_FOOT_ACTION = _mcu_config_dict["left_foot_action"]
  132. LEFT_FOOT_PHOTOGRAPH = _mcu_config_dict["left_foot_photograph"]
  133. LEFT_FOOT_ACTION_1 = _mcu_config_dict["left_foot_action_1"]
  134. LEFT_FOOT_ACTION_2 = _mcu_config_dict["left_foot_action_2"]
  135. RIGHT_FOOT_ACTION = _mcu_config_dict["right_foot_action"]
  136. RIGHT_FOOT_PHOTOGRAPH = _mcu_config_dict["right_foot_photograph"]
  137. RIGHT_FOOT_ACTION_1 = _mcu_config_dict["right_foot_action_1"]
  138. RIGHT_FOOT_ACTION_2 = _mcu_config_dict["right_foot_action_2"]
  139. NEXT_STEP = int(get_dict_value(_config_mcu_config, "next_step", 6)) # 下一步
  140. MOVE_UP = _mcu_config_dict["move_up"]
  141. MOVE_DOWN = _mcu_config_dict["move_down"]
  142. STOP = _mcu_config_dict["stop"]
  143. DOMAIN = (
  144. "https://dev2.valimart.net"
  145. if config.get("app", "url_mode") != "dev"
  146. else "https://dev2.pubdata.cn"
  147. )
  148. Company = "惠利玛"
  149. is_test_plugins = true_divide
  150. OUT_PIC_MODE = "."+getSysConfigs("basic_configs", "image_out_format", "png") # ".png"
  151. OUT_PIC_SIZE = int(getSysConfigs("basic_configs", "main_image_size", "800")) # 主图大小
  152. Mode = getSysConfigs("other_configs", "product_type", "鞋类") # 程序执行类
  153. OUT_PIC_FACTOR = float(
  154. getSysConfigs("basic_configs", "image_sharpening", "1")
  155. ) # 图片锐化
  156. RESIZE_IMAGE_MODE = 1
  157. GRENERATE_MAIN_PIC_BRIGHTNESS = 254 # 色阶是否调整到位判断
  158. RUNNING_MODE = getSysConfigs("other_configs", "running_mode", "普通模式")
  159. DEFAULT_CUTOUT_MODE = getSysConfigs("other_configs", "cutout_mode", "普通抠图")
  160. CUTOUT_MODE = 0 if getSysConfigs("other_configs", "cutout_mode", "普通抠图") =="普通抠图" else 1
  161. import importlib
  162. # plugins = [
  163. # # "custom_plugins.plugins.detail_template.huilima.detail_huilima1.DetailPicGet",
  164. # ".custom_plugins.plugins.detail_template.huilima.detail_huilima1.DetailPicGet",
  165. # # "custom_plugins.plugins.detail_template.huilima.detail_huilima2.DetailPicGet",
  166. # # "custom_plugins.plugins.detail_template.huilima.detail_huilima3.DetailPicGet",
  167. # # "custom_plugins.plugins.detail_template.huilima.detail_huilima4.DetailPicGet",
  168. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie1",
  169. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie2",
  170. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie3",
  171. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie4",
  172. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie5",
  173. # # "custom_plugins.plugins.detail_template.xiaosushuoxie.detail_xiaosushuoxie6",
  174. # # "custom_plugins.plugins.detail_template.hongqingting.detail_hongqingting1.DetailPicGet",
  175. # # "custom_plugins.plugins_mode.detail_generate_base.DetailBase",
  176. # # "custom_plugins.plugins_mode.pic_deal.PictureProcessing",
  177. # ]
  178. # def load_plugin(plugin_path):
  179. # module_name, class_name = plugin_path.rsplit(".", 1)
  180. # module = importlib.import_module(module_name)
  181. # return getattr(module, class_name)
  182. # loaded_plugins = [load_plugin(p.lstrip(".")) for p in plugins]