You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

tasks_loop.py 8.4KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417
  1. import asyncio
  2. import time
  3. import redis
  4. import os
  5. import copy
  6. import json
  7. import traceback
  8. import inspect
  9. import atexit
  10. '''
  11. umod
  12. # '''
  13. p=print
  14. async def anop(*a,**kw):pass
  15. def nop(*a,**kw):pass
  16. class Edict(dict):pass
  17. class Eobj():pass
  18. def connect_redis():
  19. return redis.Redis(host='localhost', port=6379, db=0,charset="utf-8", decode_responses=True, encoding_errors="backslashreplace")
  20. def eclass_factory(n):
  21. ret = []
  22. for k in range(n):
  23. class nx:pass
  24. # _rcls_name
  25. ret.append(nx)
  26. return ret
  27. '''
  28. # '''
  29. def callable_helper(fn):
  30. ret = {
  31. "is_callable":0,
  32. "is_coroutine":0,
  33. }
  34. ret["is_callable"]= callable(fn)
  35. ret["is_coroutine"]= inspect.iscoroutinefunction(fn)
  36. # if (ca)
  37. return ret
  38. verbose = print
  39. class DbgMixin:
  40. async def ws_rec(self,text_data,*a,**kw):
  41. # if "dbg" in
  42. text_data_json = json.loads(text_data)
  43. if not 'dbg' in text_data_json:
  44. return
  45. p("DBG!")
  46. exec(text_data_json["dbg"])
  47. class AconMixin(DbgMixin):
  48. # class AconMixin:
  49. async def connect(self):
  50. # print("-----------CONNECTING async def connect")
  51. p("ACON!!!!!!!!!!!!!!!!!!!!!!!!!!!")
  52. p(flush=True)
  53. await self.accept()
  54. verbose("<MRO")
  55. verbose(type(self))
  56. await self.call_all_mro("ws_conn0")
  57. await self.call_all_mro("ws_conn")
  58. await self.call_all_mro("ws_conn2")
  59. verbose("/MRO>")
  60. # await self.ws_conn_once(event)
  61. await self.ws_conn_once()
  62. async def ws_conn_once(self, *a,**kw):pass
  63. async def ws_disconn_once(self, *a,**kw):pass
  64. async def ws_rec_once(self, *a,**kw):pass
  65. async def receive(self, *a,**kw):
  66. print("REC")
  67. print("",end="",flush=True)
  68. # print("-----------REC")
  69. await self.call_all_mro("ws_rec",*a,**kw)
  70. await self.ws_rec_once(*a,**kw)
  71. async def call_all_mro(self,mthd_name,*args,**kwargs):
  72. called = set([None])
  73. for cls_obj in self.__class__.__mro__:
  74. mthd = getattr(cls_obj,mthd_name,None)
  75. # print("~",cls_obj,mthd)
  76. if not mthd in called:
  77. # print(cls_obj,mthd)
  78. called.add(mthd)
  79. await mthd(self,*args,**kwargs)
  80. async def disconnect(self, close_code):
  81. await self.call_all_mro("ws_disconn",close_code)
  82. await self.ws_disconn_once(close_code)
  83. async def websocket_connect_x(self, event):
  84. await super().websocket_connect(event)
  85. print("<MRO")
  86. print(type(self))
  87. await self.call_all_mro("ws_conn",event)
  88. await self.call_all_mro("ws_conn2",event)
  89. print("/MRO>")
  90. await self.ws_conn_once(event)
  91. class AclTaskMixin:
  92. async def ws_conn0(self):
  93. print("AclTaskMixin:ws_conn0")
  94. glob_cl.run_persistent_loop(self)
  95. glob_cl.acl = self.channel_layer
  96. # print("++++++++++++++++++++++++++")
  97. async def ws_conn_once(self,*a,**kw):
  98. print("\t\tws_conn_once:","AclTaskMixin")
  99. class DevMixin(*eclass_factory(10)):pass
  100. class DevMroMixin(*eclass_factory(10)):pass
  101. class WSStoMixin(*eclass_factory(10)):pass
  102. class GCLMixin(*eclass_factory(10)):pass
  103. class RldManMixin(*eclass_factory(10)):pass
  104. class LoneMixinA():pass
  105. class LoneMixinB():pass
  106. Mixins = Eobj()
  107. Mixins.DevMixin = DevMixin
  108. Mixins.DevMroMixin = DevMroMixin
  109. Mixins.WSStoMixin = WSStoMixin
  110. Mixins.GCLMixin = GCLMixin
  111. Mixins.RldManMixin = RldManMixin
  112. Mixins.LoneMixinA = LoneMixinA
  113. Mixins.LoneMixinB = LoneMixinB
  114. Mixins.DbgMixin = DbgMixin
  115. Mixins.AconMixin = AconMixin
  116. Mixins.AclTaskMixin = AclTaskMixin
  117. # GCLMixin
  118. class globCL(GCLMixin):
  119. def __init__(self,*a,**kw):
  120. # for now we are not going to do any weird class stuff
  121. self.acl = None
  122. self.rconn = connect_redis()
  123. self.rman = RldMan()
  124. self.cbs_once = []
  125. self.cbs_once_set = set()
  126. self.cbs = {"fn1":fn1,"fn2x":nop}
  127. self.timeout = 1
  128. pass
  129. async def persistent_loop(self,*a,**kw):
  130. while 1:
  131. # p("PL:",time.time(),flush=1)
  132. # await asyncio.sleep(5)
  133. # for k,v in cls.cbs.items():
  134. for k,v in [*self.cbs.items()]:
  135. try:
  136. call_info = callable_helper(v)
  137. if (call_info["is_callable"]):
  138. if call_info["is_coroutine"]:
  139. pass
  140. await v({k,self})
  141. else:
  142. v({k,self})
  143. else:
  144. pass
  145. # p(k,v)
  146. pass
  147. except Exception as e:
  148. p("persistent_loop Exception",e)
  149. p(flush=True)
  150. # raise
  151. else:
  152. pass
  153. finally:
  154. pass
  155. # cbs_once
  156. cbs_once = [*self.cbs_once,*self.cbs_once_set]
  157. self.cbs_once = []
  158. self.cbs_once_set = set()
  159. k=0
  160. for v in cbs_once:
  161. try:
  162. call_info = callable_helper(v)
  163. if (call_info["is_callable"]):
  164. if call_info["is_coroutine"]:
  165. pass
  166. await v({k,self})
  167. else:
  168. v({k,self})
  169. else:
  170. pass
  171. # p(k,v)
  172. pass
  173. except Exception as e:
  174. p("persistent_loop Exception cbs_once:",e)
  175. p(flush=True)
  176. # raise
  177. else:
  178. pass
  179. finally:
  180. pass
  181. k += 1
  182. await asyncio.sleep(self.timeout)
  183. def _run_persistent_loop(self,*a,**kw):
  184. p("_run_persistent_loop:")
  185. loop = asyncio.get_event_loop()
  186. self.run_persistent_loop = nop
  187. self.rman.persistent_loop_cb()
  188. loop.create_task(self.persistent_loop())
  189. self.loop = loop
  190. run_persistent_loop = _run_persistent_loop
  191. # RldManMixin
  192. class RldMan(RldManMixin):
  193. def __init__(self,*a,**kw):
  194. self.files = {}
  195. z="ABC"
  196. self.scopes = {
  197. "dflt_scope":{"globals":globals(),"locals":locals()},
  198. "dflt_scope_zloc":{"globals":globals(),"locals":{}},
  199. "dflt_scope_gscope":{"globals":globals(),"locals":{}},
  200. "dflt_scope_copy": {"globals":copy.copy(globals()),"locals":copy.copy(locals())},
  201. # "dflt_scope":{"globals":globals(),"locals":{}}
  202. }
  203. self.scopes['current_scope'] = self.scopes['dflt_scope_zloc']
  204. # self.scope_opt = "locals"
  205. self.scope_opt = "globals"
  206. self.print_tb = 0
  207. base_path = "/usr/games/repos/ign4/py_rld/djc/rdir1/"
  208. file_list = [
  209. base_path+"i0.py",
  210. base_path+"i1.py",
  211. base_path+"i2.py",
  212. base_path+"i3.py",
  213. base_path+"i4.py",
  214. base_path+"i5.py",
  215. ]
  216. self.add_files(file_list)
  217. # self.add_files(file_list,{"run"})
  218. def persistent_loop_cb(self,*a,**kw):
  219. base_path = "/usr/games/repos/ign4/py_rld/djc/rdir1/"
  220. file_list = [
  221. base_path+"r0.py",
  222. base_path+"r1.py",
  223. base_path+"r2.py",
  224. base_path+"r3.py",
  225. base_path+"r4.py",
  226. base_path+"r5.py",
  227. ]
  228. self.add_files(file_list)
  229. def add_files(self,files):
  230. for file in files:
  231. if type(file)==str:
  232. self.add_file(file)
  233. elif type(file)==list:
  234. self.add_file(*file)
  235. else:
  236. p("add files???",file)
  237. def add_file(self,file_name,fnx={}):
  238. self.files[file_name] = {"ftxt":"",**fnx}
  239. def get_scope(self,rfile_obj,file_name):
  240. return self.scopes["current_scope"]
  241. return {
  242. "scope":self.scopes["current_scope"],
  243. }
  244. # rfile_obj.get("scope")
  245. # if
  246. def rld_file(self,file_name,ret= {"errs":{},"all":{},"alle":{},},**kw):
  247. if file_name in self.files:
  248. rfile_obj = self.files[file_name]
  249. st = os.stat(file_name)
  250. st_tuple = (st.st_mtime,st.st_size)
  251. # if rfile_obj["ftxt"] == "":
  252. eflag ="nd"
  253. # rfile_obj["ftxt"] = st_tuple
  254. if rfile_obj["ftxt"] != st_tuple:
  255. # p(rfile_obj["ftxt"])
  256. rfile_obj["ftxt"] = st_tuple
  257. try:
  258. f = open(file_name,"r")
  259. ftxt = f.read()
  260. # scope_key = rfile_obj.get("scope")
  261. # p(scope_key,rfile_obj)
  262. f.close()
  263. scope_obj = self.get_scope(file_name,rfile_obj)
  264. if self.scope_opt == "locals":
  265. eflag ="locals"
  266. exec(ftxt,scope_obj["globals"],scope_obj["locals"])
  267. elif self.scope_opt == "globals":
  268. eflag ="globals"
  269. exec(ftxt,scope_obj["globals"])
  270. else:
  271. eflag ="[]"
  272. exec(ftxt)
  273. except Exception as e:
  274. p()
  275. print("file:",file_name)
  276. print("EXCEPT",eflag,e)
  277. if self.print_tb:
  278. traceback.print_tb(e.__traceback__,file=sys.stdout)
  279. else:
  280. pass
  281. # print("ELSE")
  282. finally:
  283. # print("FINALLY")
  284. pass
  285. print(end="",flush=True)
  286. return ret
  287. def rld_files(self):
  288. # p()
  289. ret = {
  290. "errs":{},
  291. "all":{},
  292. "alle":{},
  293. }
  294. # for k in self.files:
  295. for k in [*self.files]:
  296. self.rld_file(k,ret)
  297. # p("rld_files!")
  298. p("",end="",flush=True)
  299. return ret
  300. def fn1(*a,**kw):
  301. # rld_files
  302. glob_cl.rman.rld_files()
  303. glob_cl = globCL()
  304. glob_cl.rman.rld_files()
  305. # print("..",end="\n",flush=True)
  306. def redis_info():
  307. rconn = glob_cl.rconn
  308. keys = rconn.keys()
  309. for k in keys:
  310. p(":",k)
  311. p("len(keys)",len(keys))
  312. print("",end="",flush=True)
  313. # rconn.flushall()
  314. redis_info()
  315. r"""
  316. def atexit_fn():
  317. pass
  318. # rconn.flushall()
  319. print("ATEXIT FN")
  320. rinfo_len()
  321. keys = rconn.keys()
  322. print(keys)
  323. print(len(keys))
  324. print("t 000")
  325. # time.sleep(10)
  326. print("t 010")
  327. # time.sleep(20)
  328. print("t 020")
  329. print("ATEXIT FN...")
  330. rconn.flushall()
  331. atexit.register(atexit_fn)
  332. # """
  333. def atexit_fn():
  334. print("atexit_fn!\n\n")
  335. print("",end="",flush=True)
  336. atexit.register(atexit_fn)