python密码管理系统源码

【实例截图】

【核心代码】

import json

from django.http import JsonResponse
from django.forms.models import model_to_dict
from django.shortcuts import render, get_object_or_404

from password.models import Table
from password.backends.crypto import RsaCrypto


def table(request):
    if request.method == 'GET':
        _lists = Table.objects.all().order_by('id')

        return render(request, 'password/table.html', {'lPage': _lists})


def change_table(request):
    if request.method == 'GET':
        try:
            _data = get_object_or_404(Table, id=request.GET.get('id'))

            jsondata = model_to_dict(_data, exclude=['create_time', 'update_time'])
            return JsonResponse({'state': 1, 'message': jsondata})
        except Exception as e:
            return JsonResponse({'state': 0, 'message': 'Get Error: '   str(e)})

    if request.method == 'POST':
        print(request.body.decode('utf-8'))

        postdata = {}
        for key, value in json.loads(request.body.decode('utf-8')).items():
            # 将接收到的json数据去掉csrf和password之后的数据存入字典postdata
            if key != 'csrfmiddlewaretoken' and key != 'password':
                postdata[key] = value if value != '' else None

            # 判断当key为password时且password有值时，对password进行加密
            if key == 'password' and value:
                _m = RsaCrypto().encrypt(value)
                if _m.get('state'):
                    # 如果加密成功将postdata字典中的password换成密文
                    postdata['password'] = _m.get('message')
                else:
                    # 如果加密失败直接返回错误信息，view结束
                    return JsonResponse({'state': 0, 'message': 'Error: '   _m.get('message')})

        try:
            # 根据传入的ID判断是新建还是更新
            object, created = Table.objects.update_or_create(
                id=postdata.get('id'),
                defaults=postdata
            )

            if created:
                return JsonResponse({'state': 1, 'message': '创建成功!'})
            else:
                return JsonResponse({'state': 1, 'message': '更新成功!'})
        except Exception as e:
            return JsonResponse({'state': 0, 'message': 'Error: '   str(e)})


def delete_table(request):
    if request.method == 'POST':
        try:
            _t = Table.objects.filter(id=request.POST.get('id'))
            _t.delete()

            return JsonResponse({'state': 1, 'message': '删除成功!'})
        except Exception as e:
            return JsonResponse({'state': 0, 'message': 'Error: '   str(e)})


def decode_password(request):
    if request.method == 'GET':
        try:
            _t = get_object_or_404(Table, id=request.GET.get('id'))

            _m = RsaCrypto().decrypt(_t.password)
            if _m.get('state'):
                return JsonResponse({'state': 1, 'message': _m.get('message'),
                                     'host': '%s:%d' % (_t.host, _t.port),
                                     'username': _t.username
                                     })
            else:
                return JsonResponse({'state': 0, 'message': 'Error: '   _m.get('message')})
        except Exception as e:
            return JsonResponse({'state': 0, 'message': 'Error: '   str(e)})