valar 1.1.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

channels/consumer.py

@@ -0,0 +1,48 @@
+
+from channels.generic.websocket import AsyncJsonWebsocketConsumer
+
+VALAR_CHANNEL_GROUP = 'VALAR'
+
+class ValarConsumer(AsyncJsonWebsocketConsumer):
+
+    def __init__(self):
+        self.client = None
+        self.uid = None
+        super().__init__()
+
+    async def connect(self):
+        params = self.scope['url_route']['kwargs']
+        self.client = params.get('client')
+        await self.channel_layer.group_add(VALAR_CHANNEL_GROUP, self.channel_name)
+        await self.accept()
+
+    async def disconnect(self, code):
+        await self.channel_layer.group_discard(VALAR_CHANNEL_GROUP, self.channel_name)
+        await self.close(code)
+
+    async def receive_json(self, data, *args, **kwargs):
+        pass
+
+    async def user_emit(self, event):
+        users: list = event.get('users',[])
+        data = event.get('data',{})
+        if self.uid in users:
+            await self.send_json(data)
+
+    async def client_emit(self, event):
+        clients: list = event.get('clients',[])
+        data = event.get('data',{})
+        if self.client in clients:
+            await self.send_json(data)
+
+    async def broadcast_emit(self, event):
+        data = event.get('data',{})
+        await self.send_json(data)
+
+    async def register_emit(self, event):
+        users: list = event.get('users', [])
+        clients: list = event.get('clients',[])
+        if self.client in clients:
+            self.uid = users[0]
+
+

channels/executer.py

@@ -0,0 +1,13 @@
+import asyncio
+
+async def execute_channel(method, sender):
+    thread = asyncio.to_thread(__execute__, method, sender)
+    asyncio.create_task(thread)
+
+
+def __execute__(method, sender):
+    sender.to_clients(None, [sender.client], 'start')
+    method(sender)
+    sender.to_clients(None, [sender.client], 'stop')
+
+

channels/mapping.py

@@ -0,0 +1,23 @@
+import importlib
+
+from django.conf import settings
+from django.core.exceptions import ImproperlyConfigured
+
+from ..classes.singleton_meta import SingletonMeta
+
+class ChannelMapping(metaclass=SingletonMeta):
+    def __init__(self):
+        mapping = settings.HANDLER_MAPPING
+        root, name = mapping.rsplit('.', 1)
+        module = importlib.import_module(root)
+        if hasattr(module, name):
+            self.mapping: dict = getattr(module, name)
+        else:
+            raise ImproperlyConfigured("%r has no attribute %r" % (root, name))
+
+    def get_handler(self, handler):
+        method = self.mapping.get(handler)
+        if method is None:
+            raise ImproperlyConfigured("Cannot find handler - %r" % handler)
+        return method
+

channels/sender.py

@@ -0,0 +1,64 @@
+import json
+import time
+from datetime import datetime
+
+from asgiref.sync import async_to_sync
+from channels.layers import get_channel_layer
+from django.http import HttpRequest
+
+from .consumer import VALAR_CHANNEL_GROUP
+from ..classes.counter import Counter
+
+
+class ValarSocketSender:
+    def __init__(self, request: HttpRequest, data=None):
+        body = json.loads(request.body)
+        auth = request.headers.get('AUTH')
+        self.client = request.headers.get('CLIENT')
+        self.uid = request.session.get('UID')
+        self.handlerKey = body.get('handlerKey')
+        self.channelKey = body.get('channelKey', 'default')
+        self.data = data or body.get('data')
+        self.send = get_channel_layer().group_send
+        self.start_time = time.time()
+        if auth and not self.uid:
+            raise Exception('Unauthorized!')
+
+    @staticmethod
+    def create_counter(length:int):
+        return Counter(length)
+
+    def __convert_body(self, emit, payload, status ,clients = None, users = None):
+        return {
+            'type': emit,
+            'data': {
+                'status': status,
+                'handlerKey': self.handlerKey,
+                'channelKey': self.channelKey,
+                'payload': payload,
+                'timestamp': datetime.now().timestamp()
+            },
+            'clients': clients or [],
+            'users': users or [],
+        }
+
+
+    def to_users(self, payload,  users, status='proceed'):
+        body = self.__convert_body(emit='user.emit', payload=payload, status=status, users=users)
+        async_to_sync(self.send)(VALAR_CHANNEL_GROUP, body)
+
+    def to_clients(self,payload, clients, status='proceed', wait=False):
+        current_time = time.time()
+        time_span = current_time - self.start_time
+        if (wait and time_span > 1 and status == 'proceed') or not wait:
+            body = self.__convert_body(emit='client.emit', payload=payload, status=status, clients=clients)
+            async_to_sync(self.send)(VALAR_CHANNEL_GROUP, body)
+            self.start_time = current_time
+
+    def broadcast(self, payload, status):
+        body = self.__convert_body(emit='broadcast.emit', payload=payload, status=status)
+        async_to_sync(self.send)(VALAR_CHANNEL_GROUP, body)
+
+    def register(self):
+        body = self.__convert_body(emit='register.emit',  payload=None, status=None,clients=[self.client], users=[self.uid])
+        async_to_sync(self.send)(VALAR_CHANNEL_GROUP, body)

channels/views.py

@@ -0,0 +1,12 @@
+
+from .executer import execute_channel
+from .mapping import ChannelMapping
+from .sender import ValarSocketSender
+from ..classes.valar_response import ValarResponse
+
+
+async def handel_channel(request, handler):
+    sender = ValarSocketSender(request)
+    method = ChannelMapping().get_handler(handler)
+    await execute_channel(method, sender)
+    return ValarResponse(True)

classes/auto_migration_mixin.py

@@ -0,0 +1,7 @@
+class AutoMigrationMixin:
+    name = None  # 子类必须提供
+    def auto_migrate(self):
+        from django.core.management import call_command
+        call_command('makemigrations', self.name, interactive=False, verbosity=0)
+        call_command('migrate', self.name, interactive=False, verbosity=0)
+

classes/auto_urlpatterns_mixin.py

@@ -0,0 +1,15 @@
+import importlib
+
+from django.conf import settings
+from django.urls import path, include
+
+class AutoUrlPatternsMixin:
+    name = None  # 子类必须提供
+    def set_url(self):
+        root = settings.ROOT_URLCONF
+        module = importlib.import_module(root)
+        urlpatterns: list = getattr(module, 'urlpatterns')
+        url = f'{self.name}.urls'
+        urlpatterns.append(path('data/', include(url)))
+
+

classes/counter.py

@@ -0,0 +1,9 @@
+class Counter:
+    def __init__(self, array_or_int):
+        self.length = array_or_int if isinstance(array_or_int,int) else len(array_or_int)
+        self.index = 0
+
+    def tick(self):
+        self.index += 1
+        percentage = round(self.index * 100 / self.length) if self.length else 0
+        return {"index": self.index, "length": self.length, "percentage": percentage}

classes/singleton_meta.py

@@ -0,0 +1,6 @@
+class SingletonMeta(type):
+    _instances = {}
+    def __call__(cls, *args, **kwargs):
+        if cls not in cls._instances:
+            cls._instances[cls] = super(SingletonMeta, cls).__call__(*args, **kwargs)
+        return cls._instances[cls]

classes/valar_response.py

@@ -0,0 +1,7 @@
+from django.http import JsonResponse
+
+class ValarResponse(JsonResponse):
+    def __init__(self, data, message='', code='info'):
+        self.message = message
+        self.code = code
+        super(ValarResponse, self).__init__(data, safe=False)

{valar-1.1.0.dist-info → valar-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: valar
-Version: 1.1.0
+Version: 1.1.1
 Summary: valar for morghulis
 Home-page: https://gitee.com/GRIFFIN120/valar_dev
 Author: LYP

valar-1.1.1.dist-info/RECORD

@@ -0,0 +1,25 @@
+channels/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+channels/consumer.py,sha256=w2Cc-4Jy3tt5h8lKwtmcsJd3yo6HOIrkPspDc1dIQSg,1421
+channels/executer.py,sha256=lurFYyUETq6eWjFUXA-m7mmQTeZtwfxrzuOayWxs4BE,314
+channels/mapping.py,sha256=1SBqxEYO4epbRkSOUw5zm6gau15imz1gokrRTQcy94A,755
+channels/sender.py,sha256=x9oYOYSK0e5E37GauWF5KKbyG1toe8wRAPwSAv0TQms,2465
+channels/views.py,sha256=SMY80ZVDDjIZI-k75TymhToV-3fXPAMxcD93G7mDeRw,374
+classes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+classes/auto_migration_mixin.py,sha256=DrcZX1xR_b7S6mxvzDMidvt9uOUvyUkONlc7Dqyci2w,306
+classes/auto_urlpatterns_mixin.py,sha256=b0gB2_GpFfpLusNAFGrDvCxlSRghlNlRzNyadrtLDLo,414
+classes/counter.py,sha256=K8aQUI1RiIrSOg3280tV102soC-ZC1qSRZvBXuwNYQ0,377
+classes/singleton_meta.py,sha256=aa9DZKG4Q9Cr67ZWrQ0yaFLrEjwFoTMQbDrpoIqgJZs,245
+classes/valar_response.py,sha256=kQnHWm8McUKTE_Q20JjLQsXOAv-IX979FRg6O42Tsng,246
+frame/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+migrations/0001_initial.py,sha256=fW74vobQLZ-cJlljq7LnuxlbMb_qibtx6sVDl6Egrq8,9745
+migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+models/__init__.py,sha256=t5BRtyeM8Pa3MRpiC72IoMIl6-2YGEZrU77iemLfuSk,31
+models/core.py,sha256=a-EEN6-s_wdKrgw8nJTPvWcvxRiSG-ff7k_L5_crW-0,3439
+models/frame.py,sha256=MrgcPfBLN0nvpIm2gF68YjmavIgOMGLZUjJ-eiNb90o,1016
+models/meta.py,sha256=auy7ofa5aR_1dmp-DtoQ7DiFsqKc923-Wh0ymTh7k0A,4470
+valar-1.1.1.dist-info/licenses/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+views/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+valar-1.1.1.dist-info/METADATA,sha256=aexsxxMh3lDjfb5DclyQti1LOiJHBq4PXxRjH8NE8Ho,3394
+valar-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+valar-1.1.1.dist-info/top_level.txt,sha256=n9GN471ruasH7el7VNwZw38g_HywLvQO8AZEa90aWY8,47
+valar-1.1.1.dist-info/RECORD,,

{valar-1.1.0.dist-info → valar-1.1.1.dist-info}/top_level.txt

@@ -1,3 +1,6 @@
+channels
+classes
 frame
 migrations
 models
+views

valar-1.1.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-frame/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-migrations/0001_initial.py,sha256=fW74vobQLZ-cJlljq7LnuxlbMb_qibtx6sVDl6Egrq8,9745
-migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-models/__init__.py,sha256=t5BRtyeM8Pa3MRpiC72IoMIl6-2YGEZrU77iemLfuSk,31
-models/core.py,sha256=a-EEN6-s_wdKrgw8nJTPvWcvxRiSG-ff7k_L5_crW-0,3439
-models/frame.py,sha256=MrgcPfBLN0nvpIm2gF68YjmavIgOMGLZUjJ-eiNb90o,1016
-models/meta.py,sha256=auy7ofa5aR_1dmp-DtoQ7DiFsqKc923-Wh0ymTh7k0A,4470
-valar-1.1.0.dist-info/licenses/LICENSE,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-valar-1.1.0.dist-info/METADATA,sha256=1VudqlWI655yj_yGwO7Ap-AgyKxaZ_t9hsonBuMqZ_A,3394
-valar-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-valar-1.1.0.dist-info/top_level.txt,sha256=ePTyujihr1SPrcV-yRJAGVq34JJTyn6MmqGGvHpIGOo,24
-valar-1.1.0.dist-info/RECORD,,