智能算法后台管理

AdminController/AlgorithmShopManageController.py

@@ -6,12 +6,21 @@
 @Email   : zhangdongming@asj6.wecom.work
 @Software: PyCharm
 """
+import json
+import time
+from decimal import Decimal
+
+from django.core.paginator import Paginator
 from django.views import View
 
-from Model.models import DeviceAlgorithmExplain
+from Model.models import DeviceAlgorithmExplain, DeviceAlgorithmType
+from Object.AWS.AmazonS3Util import AmazonS3Util
 from Object.ResponseObject import ResponseObject
 from Object.TokenObject import TokenObject
 
+from Ansjer.config import CONFIG_INFO, CONFIG_EUR, CONFIG_US, CONFIG_TEST, CONFIG_CN, AWS_ACCESS_KEY_ID, \
+    AWS_SECRET_ACCESS_KEY, AWS_SES_ACCESS_REGION
+
 
 class AlgorithmShopManageView(View):
     def get(self, request, *args, **kwargs):
@@ -34,18 +43,19 @@ class AlgorithmShopManageView(View):
 
         ''' 后台管理'''
         response = ResponseObject(returntype='pc')
-        if operation == 'save':
-            pass
         if operation == 'update':
             return self.algorithm_update(request_dict, response, api_version)
+        elif operation == 'query':
+            return self.algorithm_query(request_dict, response)
+        elif operation == 'save':
+            return self.algorithm_save(request, request_dict, response)
+        elif operation == 'delete':
+            return self.algorithm_delete(request_dict, response)
+        elif operation == 'edit':
+            return self.algorithm_edit(request, request_dict, response)
         else:
             return response.json(404)
 
-    @classmethod
-    def algorithm_add(cls, request_dict, response):
-        pass
-        return response.json(0)
-
     @classmethod
     def algorithm_update(cls, request_dict, response, api_version):
         try:
@@ -80,3 +90,233 @@ class AlgorithmShopManageView(View):
         except Exception as e:
             print(repr(e))
             return response.json(500)
+
+    @classmethod
+    def algorithm_query(cls, request_dict, response):
+        algorithm_type = request_dict.get('algorithmType', None)
+        tag = request_dict.get('tag', None)
+        status = request_dict.get('status', None)
+        page = request_dict.get('page', 1)
+        page_size = request_dict.get('pageSize', 10)
+        try:
+            device_algorithm_type_qs = DeviceAlgorithmType.objects.all()
+            if algorithm_type:
+                device_algorithm_type_qs = device_algorithm_type_qs.filter(type=algorithm_type)
+            if tag:
+                device_algorithm_type_qs = device_algorithm_type_qs.filter(tag=tag)
+            if status:
+                device_algorithm_type_qs = device_algorithm_type_qs.filter(status=status)
+            # 分页
+            paginator = Paginator(device_algorithm_type_qs.order_by("-sort"), page_size)  # 每页显示 page_size 条
+            device_algorithm_type_page = paginator.get_page(page)  # 获取当前页的数据
+
+            device_algorithm_type_list = []
+            for device_algorithm_type in device_algorithm_type_page:
+                device_algorithm_explain_qs = DeviceAlgorithmExplain.objects.filter(
+                    algorithm_type_id=device_algorithm_type.id)
+                device_algorithm_explain_list = []
+                for device_algorithm_explain in device_algorithm_explain_qs:
+                    device_algorithm_explain_list.append(
+                        {
+                            'algorithmExplainId': device_algorithm_explain.id,
+                            'lang': device_algorithm_explain.lang,
+                            'title': device_algorithm_explain.title,
+                            'subtitle': device_algorithm_explain.subtitle,
+                            'price': device_algorithm_explain.price,
+                            'introduction': device_algorithm_explain.introduction,
+                            'installExplain': device_algorithm_explain.install_explain,
+                            'concerning': device_algorithm_explain.concerning,
+                            'riskWarning': device_algorithm_explain.risk_warning,
+                        }
+                    )
+                device_algorithm_explain_qs = DeviceAlgorithmExplain.objects.filter(
+                    algorithm_type_id=device_algorithm_type.id)
+                translation_no = device_algorithm_explain_qs.count()
+                device_algorithm_explain = device_algorithm_explain_qs.filter(lang="cn").first()
+                device_algorithm_type_list.append({
+                    "algorithmId": device_algorithm_type.id,
+                    "title": device_algorithm_explain.title if device_algorithm_explain.title else "",
+                    "algorithmType": device_algorithm_type.type,
+                    "memory": device_algorithm_type.memory,
+                    "downCount": device_algorithm_type.down_count,
+                    "tag": device_algorithm_type.tag,
+                    "status": device_algorithm_type.status,
+                    "expireTime": device_algorithm_type.expire_time if device_algorithm_type.status == 1 else 0,
+                    "sort": device_algorithm_type.sort,
+                    "basicFunction": device_algorithm_type.basic_function,
+                    "imageUrl": device_algorithm_type.image_url,
+                    "detailsImgUrl": device_algorithm_type.details_img_url,
+                    "iconUrl": device_algorithm_type.icon_url,
+                    "resource": device_algorithm_type.resource,
+                    "deviceAlgorithmExplain": device_algorithm_explain_list,
+                    "translationNo": translation_no
+                })
+            return response.json(0, device_algorithm_type_list)
+        except Exception as e:
+            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
+
+    @classmethod
+    def algorithm_save(cls, request, request_dict, response):
+        algorithm_type = request_dict.get('algorithmType', None)
+        memory = request_dict.get('memory', "")
+        tag = request_dict.get('tag', 0)
+        status = request_dict.get('status', 0)
+        expire_time = request_dict.get('expireTime', 0)
+        sort = request_dict.get('sort', 0)
+        basic_function = request_dict.get('basicFunction', '')
+        resource = request_dict.get('resource', None)
+        lang_configs = request_dict.get('langConfig', None)
+        image_file = request.FILES.get('imageFile', None)
+        details_img_file = request.FILES.get('detailsImgFile', None)
+        icon_file = request.FILES.get('iconFile', None)
+        if not algorithm_type:
+            return response.json(173)
+
+        try:
+            if lang_configs:
+                lang_configs = json.loads(lang_configs)
+            else:
+                lang_configs = []
+
+            device_algorithm_type_qs = DeviceAlgorithmType.objects.filter(type=algorithm_type)
+            if device_algorithm_type_qs.exists():
+                return response.json(174)
+
+            # 上传图片至存储桶
+            icon_url = cls.upload_image("app/algorithm-shop/icon/", algorithm_type, icon_file)
+            image_url = cls.upload_image("app/algorithm-shop/image/", algorithm_type, image_file)
+            details_img_url = cls.upload_image("app/algorithm-shop/details-img/", algorithm_type, details_img_file)
+
+            # 激活时间
+            expire_time = expire_time if status == 1 else 0
+
+            # 创建算法类型
+            device_algorithm_type = DeviceAlgorithmType.objects.create(type=algorithm_type, memory=memory, tag=tag,
+                                                                       status=status, expire_time=expire_time,
+                                                                       sort=sort,
+                                                                       basic_function=basic_function,
+                                                                       image_url=image_url,
+                                                                       details_img_url=details_img_url,
+                                                                       icon_url=icon_url,
+                                                                       resource=resource, created_time=int(time.time()))
+
+            # 处理算法翻译数据
+            explain_instances = []
+            for lang_config in lang_configs:
+                explain_instance = DeviceAlgorithmExplain(
+                    algorithm_type_id=int(device_algorithm_type.id),  # 关联的 DeviceAlgorithmType 的 id
+                    title=lang_config['title'],
+                    subtitle=lang_config['subtitle'],
+                    price=Decimal(lang_config['price'] if lang_config['price'] else 0),
+                    introduction=lang_config['introduction'],
+                    install_explain=lang_config['installExplain'],
+                    concerning=lang_config['concerning'],
+                    risk_warning=lang_config['riskWarning'],
+                    lang=lang_config['lang'],
+                    updated_time=int(time.time()),
+                    created_time=int(time.time())
+                )
+                explain_instances.append(explain_instance)
+            DeviceAlgorithmExplain.objects.bulk_create(explain_instances)
+            return response.json(0)
+        except Exception as e:
+            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
+
+    @classmethod
+    def algorithm_delete(cls, request_dict, response):
+        algorithm_id = request_dict.get('algorithmId', None)
+        if not algorithm_id:
+            return response.json(444)
+        DeviceAlgorithmExplain.objects.filter(algorithm_type_id=algorithm_id).delete()
+        DeviceAlgorithmType.objects.filter(id=algorithm_id).delete()
+        return response.json(0)
+
+    @classmethod
+    def upload_image(cls, directory, algorithm_type, image_file):
+        if not image_file:
+            return ""
+        bucket = 'ansjerfilemanager'
+        file_key = directory + f"{CONFIG_INFO}_{algorithm_type}.png"
+        if CONFIG_INFO == CONFIG_TEST or CONFIG_INFO == CONFIG_CN or CONFIG_INFO == 'local':
+            prefix_url = "https://ansjerfilemanager.s3.cn-northwest-1.amazonaws.com.cn/"
+            s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], 'cn-northwest-1')
+        else:
+            prefix_url = "https://ansjerfilemanager.s3.amazonaws.com/"
+            s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[1], AWS_SECRET_ACCESS_KEY[1], AWS_SES_ACCESS_REGION)
+        s3.upload_file_obj(
+            bucket,
+            file_key,
+            image_file,
+            {'ContentType': image_file.content_type, 'ACL': 'public-read'})
+        return prefix_url + file_key
+
+    @classmethod
+    def algorithm_edit(cls, request, request_dict, response):
+        algorithm_id = request_dict.get('algorithmId', None)
+        memory = request_dict.get('memory', '')
+        tag = request_dict.get('tag', 0)
+        status = request_dict.get('status', 0)
+        expire_time = request_dict.get('expireTime', 0)
+        sort = request_dict.get('sort', 0)
+        basic_function = request_dict.get('basicFunctions', '')
+        resource = request_dict.get('resource', None)
+        lang_configs = request_dict.get('langConfig', None)
+        image_file = request.FILES.get('imageFile', None)
+        details_img_file = request.FILES.get('detailsImgFile', None)
+        icon_file = request.FILES.get('iconFile', None)
+        if not algorithm_id:
+            return response.json(444)
+
+        try:
+            device_algorithm_type = DeviceAlgorithmType.objects.get(pk=algorithm_id)
+            if not device_algorithm_type:
+                return response.json(173)
+            algorithm_type = device_algorithm_type.type
+            if lang_configs:
+                lang_configs = json.loads(lang_configs)
+                DeviceAlgorithmExplain.objects.filter(algorithm_type_id=algorithm_id).delete()
+                # 处理算法翻译数据
+                explain_instances = []
+                for lang_config in lang_configs:
+                    explain_instance = DeviceAlgorithmExplain(
+                        algorithm_type_id=int(algorithm_id),  # 关联的 DeviceAlgorithmType 的 id
+                        title=lang_config['title'],
+                        subtitle=lang_config['subtitle'],
+                        price=Decimal(lang_config['price'] if lang_config['price'] else 0),
+                        introduction=lang_config['introduction'],
+                        install_explain=lang_config['installExplain'],
+                        concerning=lang_config['concerning'],
+                        risk_warning=lang_config['riskWarning'],
+                        lang=lang_config['lang'],
+                        updated_time=int(time.time()),
+                        created_time=int(time.time())
+                    )
+                    explain_instances.append(explain_instance)
+                DeviceAlgorithmExplain.objects.bulk_create(explain_instances)
+            if memory:
+                device_algorithm_type.memory = memory
+            if tag:
+                device_algorithm_type.tag = tag
+            if status:
+                device_algorithm_type.status = status
+                if status == 1 and expire_time:
+                    device_algorithm_type.expire_time = expire_time
+            if sort:
+                device_algorithm_type.sort = sort
+            if basic_function:
+                device_algorithm_type.basic_function = basic_function
+            if resource:
+                device_algorithm_type.resource = resource
+            if image_file:
+                image_url = cls.upload_image("app/algorithm-shop/image/", algorithm_type, image_file)
+                device_algorithm_type.image_url = image_url
+            if icon_file:
+                icon_url = cls.upload_image("app/algorithm-shop/icon/", algorithm_type, icon_file)
+                device_algorithm_type.icon_url = icon_url
+            if details_img_file:
+                details_img_url = cls.upload_image("app/algorithm-shop/details-img/", algorithm_type, details_img_file)
+                device_algorithm_type.details_img_url = details_img_url
+            device_algorithm_type.save()
+            return response.json(0)
+        except Exception as e:
+            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))