使用命令行 执行脚本
from django.core.management.base import BaseCommand
from django.db import transaction
from apps.db.models import AdminMenu
class Command(BaseCommand):
help = '管理端、PC端、H5端菜单'
def add_arguments(self, parser):
parser.add_argument('-func', action='store', dest='func', default=False, help='')
parser.add_argument('-arg', action='store', dest='arg', default='', help='')
def handle(self, *args, **options):
# print(args, options)
# print('====禁止执行====')
# return
_func = options.get('func', None) # 操作方法
arg = options.get('arg', None) # 操作参数
if hasattr(self, _func):
if arg:
eval('self.%s(%s)' % (_func, arg))
else:
eval('self.%s()' % _func)
else:
print('error: 无效的参数func')
print('===========end==============')
def pc(self):
menus = [
{"name": "主页",
"code": "",
"sort": 1,
"icon": "/statics/icon/admin/menu1.svg",
"is_menu": True,
"type": 2,
"subs": []
},
]
with transaction.atomic():
for menu in menus:
obj1 = AdminMenu.objects.create(
name=menu['name'],
code=menu['code'],
sort=menu['sort'],
icon=menu['icon'],
is_menu=menu['is_menu'],
type=menu['type'],
enabled=True,
pid=None,
pids=None
)
self.sub_menu(obj1, menu['subs'], [obj1.uuid])
def sub_menu(self, obj1, subs, pids):
if subs:
for sub in subs:
obj2 = AdminMenu.objects.create(
name=sub['name'],
code=sub['code'],
sort=sub['sort'],
icon=sub['icon'],
is_menu=sub['is_menu'],
type=sub['type'],
enabled=True,
pid=obj1,
pids=pids
)
sub_pids = pids + [obj2.uuid]
self.sub_menu(obj2, sub['subs'], sub_pids)
用法