feat(recipe views): 增加菜谱原材料统计接口

增加菜谱原材料统计接口

Signed-off-by: Ching <loooching@gmail.com>

.gitignore

@@ -10,6 +10,7 @@ __pycache__/
 dsite/settings.py
 
 # Distribution / packaging
+dump.rdb
 .Python
 build/
 develop-eggs/

dsite/settings_default.py

@@ -148,3 +148,9 @@ IG_LOGIN_USERNAME = ''
 IG_LOGIN_PASSWORD = ''
 
 MASTODON_NOFAN_ACCESS_TOKEN = ''
+
+LARK_APP_ID = ''
+LARK_APP_SECRET = ''
+
+FANFOU_USERNAME = ''
+FANFOU_PASSWORD = ''

recipe/models.py

@@ -141,6 +141,22 @@ class Recipe(models.Model):
         }
         return data
 
+    @classmethod
+    def sum_recipe_ingredients(cls, recipe_ids):
+        data = {}
+        for recipe_id in recipe_ids:
+            recipe = cls.objects.get(id=recipe_id)
+            recipe_ingredients = RecipeIngredient.objects.filter(recipe=recipe)
+            for ingredient in recipe_ingredients:
+                if not ingredient.ingredient.id in data:
+                    data[ingredient.ingredient.id] = {
+                        'name': ingredient.ingredient.name,
+                        'quantity': 0,
+                        'unit': ingredient.ingredient.unit,
+                    }
+                data[ingredient.ingredient.id]['quantity'] += ingredient.quantity
+        return data
+
 
 class DailyRecipe(models.Model):
     recipes = models.ManyToManyField(Recipe)

recipe/serializers.py

@@ -58,6 +58,10 @@ class RecipeSerializer(serializers.ModelSerializer):
         fields = ('recipe_ingredients', 'id', 'name', 'recipe_type', 'status', 'note', 'rate', 'difficulty')
 
 
+class RecipeIngredientSummarySerializer(serializers.Serializer):
+    pass
+
+
 class WeekRecipeSerializer(serializers.ModelSerializer):
     class Meta:
         model = recipe.models.DailyRecipe

recipe/urls.py

@@ -17,6 +17,7 @@ recipe_nested_router.register(r'recipe-ingredient', views.RecipeIngredientAPI)
 urlpatterns = [
     url(r'^week-recipe/$', views.WeekRecipeListAPI.as_view()),
     url(r'^daily-recipe/(?P<pk>\d+)$', views.DailyRecipeAPI.as_view(), name='dailyrecipe-detail'),
+    url(r'^ingredient-summary/$', views.RecipeIngredientSummaryAPI.as_view()),
     path(r'', include(router.urls)),
     path(r'', include(recipe_nested_router.urls)),
 ]

recipe/views.py

@@ -142,3 +142,24 @@ class RecipeIngredientAPI(viewsets.ModelViewSet):
             recipe_id=kwargs['recipe_pk'], ingredient_id=request.data['ingredient'], quantity=request.data['quantity']
         )
         return Response(recipe_ingredient.serialize(), status=status.HTTP_201_CREATED, headers={})
+
+
+# class RecipeIngredientSummaryAPI(viewsets.ViewSetMixin, rest_framework.generics.CreateAPIView):
+class RecipeIngredientSummaryAPI(rest_framework.generics.CreateAPIView):
+
+    # authentication_classes = (authentication.TokenAuthentication,
+    #                           authentication.SessionAuthentication,
+    #                           authentication.BasicAuthentication)
+    # permission_classes = (permissions.IsAuthenticated,)
+    queryset = recipe.models.RecipeIngredient.objects.all()
+    serializer_class = recipe.serializers.RecipeIngredientSummarySerializer
+
+    def post(self, request, *args, **kwargs):
+        daily_recipe_ids = request.data.get('daily_recipes', [])
+        recipe_ids = request.data.get('recipes', [])
+        if daily_recipe_ids:
+            daily_recipes = recipe.models.DailyRecipe.objects.filter(id__in=daily_recipe_ids)
+            recipe_ids = [recipe.id for daily_recipe in daily_recipes for recipe in daily_recipe.recipes.all()]
+
+        data = recipe.models.Recipe.sum_recipe_ingredients(recipe_ids)
+        return Response(data, status=status.HTTP_200_OK, headers={})

scripts/dodo.py

@@ -26,23 +26,23 @@ from bs4 import BeautifulSoup
 
 import recipe.models
 from utils import const
+import utils.feishu
 
 APP_VERIFICATION_TOKEN = 'uKQQiOVMYg2cTgrjkyBmodrHTUaCXzG3'
-APP_ID = 'cli_a115fe8b83f9100c'
-APP_SECRET = 'yuSQenId0VfvwdZ3qL9wMd8FpCMEUL0u'
 ENCRYPT_KEY = '4XfjcA5xou3pztBD4g5V7dgHtr0BBYDE'
 EVENT_TYPE = ['im.message.receive_v1']
 ADD_GROUP_NAME = True
 KEDAI_ID = '107263380636355825'
 
-logging.basicConfig(filename='/root/develop/log/dodo.log', level=logging.INFO)
+logger = logging.getLogger('dsite.' + __name__)
-logger = logging.getLogger('/root/develop/log/dodo.log')
 
 mastodon_cli = Mastodon(access_token='Ug_bUMWCk3RLamOnqYIytmeB0nO6aNfjdmf06mAj2bE', api_base_url='https://nofan.xyz')
 
 pool = redis.ConnectionPool(host='localhost', port=6379, decode_responses=True)
 redis_cli = redis.Redis(host='localhost', port=6379, decode_responses=True)
 
+feishu_cli = utils.feishu.Feishu()
+
 
 class AESCipher(object):
     def __init__(self, key):
@@ -302,7 +302,8 @@ class RequestHandler(BaseHTTPRequestHandler):
             logger.error("send message error, code = %s, msg =%s", code, rsp_dict.get("msg", ""))
 
     def msg_compoment(self, token, open_id, text, msg_type=None, content=None):
-        self.send_message(token, open_id, text, msg_type, content)
+        # self.send_message(token, open_id, text, msg_type, content)
+        feishu_cli.send_message(open_id, text, msg_type, content)
 
 
 def run():

utils/feishu.py

@@ -0,0 +1,94 @@
+import redis
+import requests
+from urllib import request
+import json
+import logging
+
+from dsite import settings
+from utils import const
+
+# logging.basicConfig(filename='/root/develop/log/dodo.log', level=logging.INFO)
+# logger = logging.getLogger('/root/develop/log/dodo.log')
+
+logger = logging.getLogger('dsite.' + __name__)
+
+APP_ID = settings.LARK_APP_ID
+APP_SECRET = settings.LARK_APP_SECRET
+
+pool = redis.ConnectionPool(host='localhost', port=6379, decode_responses=True)
+redis_cli = redis.Redis(host='localhost', port=6379, decode_responses=True)
+
+
+class Feishu:
+    def __init__(self):
+        self.token = self.get_tenant_access_token()
+
+    def get_tenant_access_token(self):  # 获取token
+        token = redis_cli.get('tenant_access_token_%s' % APP_ID)
+        if token:
+            return token
+
+        url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal/"
+        headers = {"Content-Type": "application/json"}
+        req_body = {"app_id": APP_ID, "app_secret": APP_SECRET}
+
+        data = bytes(json.dumps(req_body), encoding='utf8')
+        req = request.Request(url=url, data=data, headers=headers, method='POST')
+        try:
+            response = request.urlopen(req)
+        except Exception as e:
+            logger.error('get tenant token error: %s', e.read().decode())
+            return ""
+
+        rsp_body = response.read().decode('utf-8')
+        rsp_dict = json.loads(rsp_body)
+        code = rsp_dict.get("code", -1)
+        if code != 0:
+            logger.error("get tenant_access_token error, code =%s", code)
+            return ""
+        token = rsp_dict.get("tenant_access_token", "")
+        redis_cli.set('tenant_access_token_%s' % APP_ID, rsp_dict.get("tenant_access_token", ""), ex=60 * 30)
+
+        return token
+
+    def send_message(self, open_id, text, msg_type=None, content=None):
+        url = "https://open.feishu.cn/open-apis/message/v4/send/"
+        headers = {"Content-Type": "application/json", "Authorization": "Bearer " + self.token}
+        if not msg_type:
+            msg_type = const.LARK_WEBHOOK_MSG_TYPE_TEXT
+        req_body = {"msg_type": msg_type}
+        if msg_type == const.LARK_WEBHOOK_MSG_TYPE_TEXT:
+            req_body['content'] = {'text': text}
+        elif msg_type == const.LARK_WEBHOOK_MSG_TYPE_INTERACTIVE:
+            req_body['card'] = content
+
+        req_body = dict(req_body, **open_id)  # 根据open_id判断返回域
+
+        data = bytes(json.dumps(req_body), encoding='utf8')
+        req = request.Request(url=url, data=data, headers=headers, method='POST')
+        try:
+            response = request.urlopen(req)
+        except Exception as e:
+            logger.error('send message error: %s', e.read().decode())
+            return
+
+        rsp_body = response.read().decode('utf-8')
+        rsp_dict = json.loads(rsp_body)
+        code = rsp_dict.get("code", -1)
+        if code != 0:
+            logger.error("send message error, code = %s, msg =%s", code, rsp_dict.get("msg", ""))
+
+    def msg_compoment(self, open_id, text, msg_type=None, content=None):
+        self.send_message(open_id, text, msg_type, content)
+
+    def send_web_hook_message(self, web_hook_url, content):
+        """发送webhook消息
+
+        Args:
+            web_hook_url (sting): web hook url
+            content (dict): 消息内容 {"msg_type":"text","content":{"text":"request example"}}
+
+        Returns:
+            resp: response object
+        """
+        return requests.post(web_hook_url, json=content)