From 8933b14b9c04aada954972749241e6d37d95d82c Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Tue, 24 Dec 2019 09:01:12 +0800
Subject: [PATCH] add menu

---
 safesite/views.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/safesite/views.py b/safesite/views.py
index aedefa18..6a3a6ccc 100644
--- a/safesite/views.py
+++ b/safesite/views.py
@@ -4564,9 +4564,9 @@ def apitool(req):
         if req.GET.get('company'):
             objs = objs.filter(usecomp__partid=req.GET.get('company'))
         for i in objs:
-            menulist = i.menulink.split(',')
+            menulist = i.menulist
             menulist.append(menu)
-            if menu == '30':
+            if menu == 30:
                 company = Partment.objects.get(partid=req.GET.get('company'))
                 objs = Map.objects.filter(usecomp=company)
                 if objs.exists():
@@ -4574,8 +4574,7 @@ def apitool(req):
                 else:
                     Map.objects.create(
                         name='默认地图', pic='/static/safesite/mystatic/images/snc.jpg', usecomp=company, default=1)
-            newlist = list(set(menulist))
-            i.menulink = ','.join(newlist)
+            i.menulist = list(set(menulist))
             i.save()
         return JsonResponse({"code": 1})
     elif a == 'correction':