transfer.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299
  1. # -*- coding: utf-8 -*-
  2. from enum import Enum
  3. from typing import Dict, List
  4. import arrow
  5. import numpy as np
  6. import pandas as pd
  7. from httpx import AsyncClient, URL
  8. from loguru import logger
  9. from app.core.config import settings
  10. from app.services.service import Service
  11. from app.utils.date import get_time_str, TIME_FMT
  12. from app.utils.math import round_half_up
  13. class Season(str, Enum):
  14. cooling = 'Cooling'
  15. heating = 'Warm'
  16. transition = 'Transition'
  17. class SpaceInfoService(Service):
  18. def __init__(
  19. self,
  20. client: AsyncClient,
  21. project_id: str,
  22. space_id: str,
  23. server_settings=settings
  24. ) -> None:
  25. super(SpaceInfoService, self).__init__(client)
  26. self._project_id = project_id
  27. self._space_id = space_id
  28. self._base_url = URL(server_settings.TRANSFER_HOST)
  29. self._now_time = get_time_str()
  30. def _common_parameters(self) -> Dict:
  31. return {'projectId': self._project_id, 'spaceId': self._space_id}
  32. async def is_customized(self) -> bool:
  33. url = self._base_url.join('duoduo-service/custom-service/custom/timetarget')
  34. time_str = arrow.get(arrow.get(self._now_time, TIME_FMT).shift(minutes=15).timestamp
  35. // 900 * 900).strftime('%Y%m%d%H%M%S')
  36. params = {
  37. 'projectId': self._project_id,
  38. 'objectId': self._space_id,
  39. 'timepoint': time_str,
  40. }
  41. raw_info = await self._get(url, params)
  42. flag = False
  43. if raw_info.get('data'):
  44. flag = True
  45. return flag
  46. async def is_temporary(self) -> bool:
  47. url = self._base_url.join('duoduo-service/transfer/environment/temp/target')
  48. params = self._common_parameters()
  49. params.update({'time': self._now_time})
  50. raw_info = await self._get(url, params)
  51. flag = False
  52. if raw_info.get('flag') == 1:
  53. flag = True
  54. return flag
  55. async def get_feedback(self, wechat_time: str) -> Dict:
  56. url = self._base_url.join('duoduo-service/transfer/environment/feedbackCount')
  57. params = self._common_parameters()
  58. params.update({'time': wechat_time})
  59. raw_info = await self._get(url, params)
  60. meaning_dict = {
  61. 'Id1': 'a little cold',
  62. 'Id2': 'so cold',
  63. 'Id3': 'a little hot',
  64. 'Id4': 'so hot',
  65. 'Id5': 'noisy or blowy',
  66. 'Id6': 'so stuffy',
  67. 'Id7': 'more sunshine',
  68. 'Id8': 'less sunshine',
  69. 'Id9': 'send a repairman',
  70. 'Id10': 'switch off',
  71. 'Id11': 'nice',
  72. 'Id12': 'switch on',
  73. }
  74. feedback_dic = {meaning_dict.get(k): v for k, v in raw_info.items() if k != 'result'}
  75. return feedback_dic
  76. async def get_custom_target(self) -> Dict[str, pd.DataFrame]:
  77. url = self._base_url.join('duoduo-service/transfer/environment/normalAndPreDayTarget')
  78. params = self._common_parameters()
  79. params.update({'date': arrow.get(self._now_time, TIME_FMT).date().strftime('%Y%m%d')})
  80. raw_info = await self._get(url, params)
  81. try:
  82. pre_target_df = pd.DataFrame(raw_info.get('perTargets'))
  83. normal_target_df = pd.DataFrame(raw_info.get('normalTargets'))
  84. pre_target_df.set_index('time', inplace=True)
  85. normal_target_df.set_index('time', inplace=True)
  86. except KeyError:
  87. pre_target_df = pd.DataFrame()
  88. normal_target_df = pd.DataFrame()
  89. return {
  90. 'pre_targets': pre_target_df,
  91. 'normal_targets': normal_target_df
  92. }
  93. async def get_current_temperature_target(self) -> float:
  94. targets = await self.get_custom_target()
  95. if len(targets.get('pre_targets')) > 0:
  96. current_targets = targets.get('pre_targets').append(targets.get('normal_targets'))
  97. else:
  98. current_targets = targets.get('normal_targets')
  99. temp = arrow.get(self._now_time, TIME_FMT).shift(minutes=15).timestamp // (15 * 60) * (15 * 60)
  100. next_quarter_minutes = arrow.get(temp).time().strftime('%H%M%S')
  101. try:
  102. current_lower_target = current_targets['temperatureMin'].loc[next_quarter_minutes]
  103. current_upper_target = current_targets['temperatureMax'].loc[next_quarter_minutes]
  104. except KeyError:
  105. current_lower_target, current_upper_target = np.NAN, np.NAN
  106. return round_half_up((current_lower_target + current_upper_target) / 2, 2)
  107. async def env_database_set(self, form: str, value: float) -> None:
  108. url = self._base_url.join('duoduo-service/transfer/environment/hispoint/set')
  109. params = self._common_parameters()
  110. time_str = arrow.get(arrow.get(self._now_time, TIME_FMT).timestamp // 900 * 900).strftime('%Y%m%d%H%M%S')
  111. params.update({'time': time_str, 'type': form, 'value': value})
  112. await self._get(url, params)
  113. async def env_database_get(self) -> Dict[str, pd.DataFrame]:
  114. url = self._base_url.join('duoduo-service/transfer/environment/hispoint/get')
  115. params = self._common_parameters()
  116. params.update({'date': arrow.get(self._now_time, TIME_FMT).date().strftime('%Y%m%d')})
  117. raw_info = await self._get(url, params)
  118. result = {}
  119. if raw_info.get('result') == 'success':
  120. for k, v in raw_info.items():
  121. if k != 'result':
  122. if len(v) > 0:
  123. temp = {}
  124. data = np.array(v)
  125. temp.update({'timestamp': data[:, 0]})
  126. temp.update({'value': data[:, 1].astype(np.float)})
  127. result.update({k: pd.DataFrame(temp)})
  128. else:
  129. result.update({k: pd.DataFrame()})
  130. return result
  131. async def set_custom_target(self, form: str, target_value: Dict[str, List[float]], flag: str = '1') -> None:
  132. url = self._base_url.join('duoduo-service/transfer/environment/target/setting')
  133. params = {
  134. 'projectId': self._project_id,
  135. 'spaceId': self._space_id,
  136. 'timepoint': self._now_time,
  137. 'type': form,
  138. 'flag': flag
  139. }
  140. await self._post(url, params=params, payload=target_value)
  141. async def set_temporary_custom(self) -> None:
  142. url = self._base_url.join('duoduo-service/transfer/environment/setServiceFlag')
  143. params = self._common_parameters()
  144. params.update({'time': self._now_time})
  145. await self._get(url, params)
  146. async def get_equipment(self) -> List[dict]:
  147. url = self._base_url.join('duoduo-service/object-service/object/equipment/findForServe')
  148. params = self._common_parameters()
  149. raw_info = await self._post(url, params)
  150. result = []
  151. for eq in raw_info.get('data'):
  152. result.append({'id': eq.get('id'), 'category': eq.get('equipmentCategory')})
  153. return result
  154. class Duoduo(Service):
  155. def __init__(self, client: AsyncClient, project_id: str, server_settings=settings):
  156. super(Duoduo, self).__init__(client)
  157. self._project_id = project_id
  158. self._base_url = URL(server_settings.TRANSFER_HOST)
  159. self._now_time = get_time_str()
  160. async def get_season(self) -> Season:
  161. url = self._base_url.join('duoduo-service/transfer/environment/getSeasonType')
  162. params = {
  163. 'projectId': self._project_id,
  164. 'date': self._now_time,
  165. }
  166. raw_info = await self._get(url, params)
  167. return Season(raw_info.get('data'))
  168. async def get_fill_rate(self):
  169. url = self._base_url.join('duoduo-service/review-service/space/report/quarter/query')
  170. payload = {
  171. 'criteria': {
  172. 'projectId': self._project_id,
  173. 'date': arrow.get(self._now_time, TIME_FMT).date().strftime('%Y%m%d')
  174. },
  175. 'orders': [
  176. {
  177. 'column': 'time',
  178. 'asc': False
  179. }
  180. ],
  181. 'page': 1,
  182. 'size': 1
  183. }
  184. raw_info = await self._post(url, payload=payload)
  185. try:
  186. content = raw_info.get('content')[-1]
  187. hot_rate = (content.get('hotNum')
  188. / (content.get('normalNum') + content.get('hotNum') + content.get('coldNum')))
  189. except KeyError and ZeroDivisionError:
  190. hot_rate = 0.0
  191. return hot_rate
  192. async def get_space_by_equipment(self, equipment_id: str) -> List[dict]:
  193. url = self._base_url.join('duoduo-service/object-service/object/space/findForServe')
  194. params = {
  195. 'projectId': self._project_id,
  196. 'objectId': equipment_id
  197. }
  198. raw_info = await self._post(url, params)
  199. result = []
  200. for sp in raw_info.get('data'):
  201. if sp.get('isControlled'):
  202. result.append({'id': sp.get('id')})
  203. return result
  204. async def get_system_by_equipment(self, equipment_id: str) -> List:
  205. url = self._base_url.join('duoduo-service/object-service/object/system/findForCompose')
  206. params = {
  207. 'projectId': self._project_id,
  208. 'equipmentId': equipment_id
  209. }
  210. raw_info = await self._post(url, params)
  211. system_list = []
  212. for sy in raw_info.get('data'):
  213. system_list.append({'id': sy.get('id')})
  214. return system_list
  215. async def get_day_type(self) -> Dict:
  216. url = self._base_url.join('duoduo-service/custom-service/custom/getDateInfo')
  217. params = {
  218. 'projectId': self._project_id,
  219. 'date': arrow.get(self._now_time, TIME_FMT).date().strftime('%Y%m%d')
  220. }
  221. raw_info = await self._get(url, params)
  222. result = {
  223. 'day_type': raw_info.get('dayType'),
  224. 'season': raw_info.get('seasonType')
  225. }
  226. return result
  227. async def query_device_virtual_data(self, device_id: str, info_code: str) -> float:
  228. url = self._base_url.join('duoduo-service/review-service/equipment/order/query')
  229. payload = {
  230. 'criteria': {
  231. 'projectId': self._project_id,
  232. 'objectId': device_id,
  233. 'date': arrow.get(self._now_time, TIME_FMT).date().strftime('%Y%m%d'),
  234. 'funcId': info_code
  235. }
  236. }
  237. raw_info = await self._post(url, payload=payload)
  238. try:
  239. latest_data = raw_info.get('data')[-1].get('value')
  240. latest_time = raw_info.get('data')[-1].get('realTime')
  241. if arrow.get(latest_time, TIME_FMT).shift(minutes=15) < arrow.get(self._now_time, TIME_FMT):
  242. value = np.NAN
  243. else:
  244. value = latest_data
  245. except KeyError:
  246. value = np.NAN
  247. return value