def save_info(): data = request.get_json(silent=True) name = data.get('name') parent_uid = data.get('parent_uid') with db.auto_commit(): Zone.save(name=name, parent_uid=parent_uid) return Success()
def remove(): data = request.get_json(silent=True) uid = data.get('uid') z = Zone.get_item_by_uid(uid) with db.auto_commit(): z.remove() return Success()
def update_info(): data = request.get_json(silent=True) uid = data.get('uid') name = data.get('name') z = Zone.get_item_by_uid(uid=uid) with db.auto_commit(): z.update(name=name) return Success()
def get_list(): zones = Zone.recursion_items() data = { 'pageSize': len(zones), 'pageNo': 1, 'totalCount': len(zones), 'data': zones } return Success(msg=data)
def get_list(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) zone_uid = request.args.get('zone_uid', None, type=str) ip = request.args.get('ip', '', type=str) origin = request.args.get('origin', '', type=str) cpe = request.args.get('cpe', '', type=str) status = request.args.get('status', 1, type=int) pagination = Host.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, ip=ip, zone_list=Zone.recursion_children_uid_list(zone_uid), origin=origin, cpe=cpe, status=status) data = { 'pageSize': pagination.per_page, 'pageNo': pagination.page, 'totalCount': pagination.total, 'data': [{ 'uid': h.uid, 'ip': h.ip, 'zone_uid': h.zone_uid, 'origin': h.origin, 'cpe': h.cpe, 'service_count': h.service_count, 'status': h.status, 'info': h.info, } for h in pagination.items], 'zones': Zone.recursion_items() } return Success(msg=data)
def get_list(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) name = request.args.get('name', '', type=str) domain_type = request.args.get('type', '', type=str) origin = request.args.get('origin', '', type=str) zone_uid = request.args.get('zone_uid', '', type=str) status = request.args.get('status', 1, type=int) pagination = Domain.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, zone_list=Zone.recursion_children_uid_list(zone_uid), name=name, type=domain_type, origin=origin, status=status) data = { 'pageSize': pagination.per_page, 'pageNo': pagination.page, 'totalCount': pagination.total, 'data': [{ 'uid': d.uid, 'name': d.name, 'type': d.type, 'origin': d.origin, 'zone_uid': d.zone_uid, 'status': d.status, 'info': d.info } for d in pagination.items], 'zones': Zone.recursion_items() } return Success(msg=data)
def deploy(): db.drop_all() db.create_all() users = [ { 'name': '*****@*****.**', 'password': '******' }, ] roles = [ { 'name': 'admin' }, ] with db.auto_commit(): for user in users: User.save(**user) for role in roles: Role.save(**role) for item in app.url_map.iter_rules(): if item.endpoint != 'static': Permission.save(endpoint=item.endpoint) with db.auto_commit(): admin_user = User.get_item_by_name(name='*****@*****.**') admin_role = Role.get_item_by_name(name='admin') admin_role.update( permissions=[permission for permission in Permission.list_items()]) admin_user.update(roles=[admin_role]) with db.auto_commit(): # Zone.save(name='test') zone = Zone.save(name='公网') Zone.save(name='私有云', parent_uid=zone.uid) Zone.save(name='公有云', parent_uid=zone.uid) Zone.save(name='内网') zone = Zone.query.filter(Zone.name == '公有云').first() import json with open('domain.txt') as f: domain_list = json.loads(f.read()) with db.auto_commit(): for domain in domain_list: Domain.save(zone_uid=zone.uid, **domain) with open('host.txt') as f: host_list = json.loads(f.read()) with db.auto_commit(): for host in host_list: Host.save(zone_uid=zone.uid, **host) with open('service.txt') as f: service_list = json.loads(f.read()) with db.auto_commit(): for service in service_list: Service.save(zone_uid=zone.uid, **service)
def get_list(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) zone_uid = request.args.get('zone_uid', '', type=str) uid = request.args.get('uid', '', type=str) host_ip = request.args.get('host_ip', '', type=str) port = request.args.get('port', '', type=str) name = request.args.get('name', '', type=str) protocol = request.args.get('protocol', '', type=str) tunnel = request.args.get('tunnel', '', type=str) cpe = request.args.get('cpe', '', type=str) status = request.args.get('status', 1, type=int) pagination = Service.list_items_paginate_by_search(page=page_no, per_page=page_size, uid=uid, host_ip=host_ip, port=port, status=status, cpe=cpe, name=name, zone_list=Zone.recursion_children_uid_list(zone_uid), protocol=protocol, tunnel=tunnel) data = { 'pageSize': pagination.per_page, 'pageNo': pagination.page, 'totalCount': pagination.total, 'data': [{ 'uid': s.uid, 'zone_uid': s.zone_uid, 'host_uid': s.host_uid, 'host_ip': s.host_ip, 'port': s.port, 'protocol': s.protocol, 'tunnel': s.tunnel, 'name': s.name, 'cpe': s.cpe, 'status': s.status, 'info': s.info } for s in pagination.items], 'zones': Zone.recursion_items() } return Success(msg=data)
def get_host(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) zone_uid = request.args.get('zone_uid', '', type=str) ip = request.args.get('ip', '', type=str) origin = request.args.get('origin', '', type=str) cpe = request.args.get('cpe', '', type=str) pagination = Host.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, ip=ip, zone_list=Zone.recursion_children_uid_list(zone_uid), origin=origin, cpe=cpe, status=1) return Success(msg=pagination)
def get_domain(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) zone_uid = request.args.get('zone_uid', '', type=str) name = request.args.get('name', '', type=str) domain_type = request.args.get('type', '', type=str) origin = request.args.get('origin', '', type=str) pagination = Domain.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, zone_list=Zone.recursion_children_uid_list(zone_uid), name=name, type=domain_type, origin=origin, status=1) return Success(msg=pagination)
def get_http(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) zone_uid = request.args.get('zone_uid', '', type=str) website = request.args.get('website', '', type=str) business = request.args.get('business', '', type=str) title = request.args.get('title', '', type=str) status = request.args.get('status', 1, type=int) pagination = HTTP.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, website=website, zone_list=Zone.recursion_children_uid_list(zone_uid), business=business, title=title, status=status) return Success(msg=pagination)
def get_service(): page_no = request.args.get('pageNo', 1, type=int) page_size = request.args.get('pageSize', 10, type=int) uid = request.args.get('uid', '', type=str) zone_uid = request.args.get('zone_uid', '', type=str) host_ip = request.args.get('host_ip', '', type=str) port = request.args.get('port', '', type=str) name = request.args.get('name', '', type=str) protocol = request.args.get('protocol', '', type=str) tunnel = request.args.get('tunnel', '', type=str) cpe = request.args.get('cpe', '', type=str) pagination = Service.list_items_paginate_by_search( page=page_no, per_page=page_size, uid=uid, host_ip=host_ip, port=port, status=1, cpe=cpe, name=name, zone_list=Zone.recursion_children_uid_list(zone_uid), protocol=protocol, tunnel=tunnel) return Success(msg=pagination)