修改加密算法

DjangoBlog/tests.py

@@ -27,7 +27,7 @@ class DjangoBlogTest(TestCase):
         pass
 
     def test_utils(self):
-        md5 = get_md5('test')
+        md5 = get_sha256('test')
         self.assertIsNotNone(md5)
         c = CommonMarkdown.get_markdown('''
         # Title1

DjangoBlog/utils.py

@@ -14,7 +14,7 @@
 """
 from django.core.cache import cache
 from django.contrib.sites.models import Site
-from hashlib import md5
+from hashlib import sha256
 import mistune
 from mistune import escape, escape_link
 from pygments import highlight
@@ -34,8 +34,8 @@ def get_max_articleid_commentid():
     return (Article.objects.latest().pk, Comment.objects.latest().pk)
 
 
-def get_md5(str):
-    m = md5(str.encode('utf-8'))
+def get_sha256(str):
+    m = sha256(str.encode('utf-8'))
     return m.hexdigest()
 
 
@@ -50,7 +50,7 @@ def cache_decorator(expiration=3 * 60):
             if not key:
                 unique_str = repr((func, args, kwargs))
 
-                m = md5(unique_str.encode('utf-8'))
+                m = sha256(unique_str.encode('utf-8'))
                 key = m.hexdigest()
             value = cache.get(key)
             if value is not None:

accounts/tests.py

@@ -65,7 +65,7 @@ class AccountTest(TestCase):
                 BlogUser.objects.filter(
                     email='user123@user.com')))
         user = BlogUser.objects.filter(email='user123@user.com')[0]
-        sign = get_md5(get_md5(settings.SECRET_KEY + str(user.id)))
+        sign = get_sha256(get_sha256(settings.SECRET_KEY + str(user.id)))
         path = reverse('accounts:result')
         url = '{path}?type=validation&id={id}&sign={sign}'.format(
             path=path, id=user.id, sign=sign)

accounts/views.py

@@ -17,7 +17,7 @@ from django.shortcuts import redirect
 from django.utils.decorators import method_decorator
 from django.views.decorators.debug import sensitive_post_parameters
 from django.utils.http import is_safe_url
-from DjangoBlog.utils import send_email, get_md5, get_current_site
+from DjangoBlog.utils import send_email, get_sha256, get_current_site
 from django.conf import settings
 
 logger = logging.getLogger(__name__)
@@ -36,7 +36,7 @@ class RegisterView(FormView):
             user.source = 'Register'
             user.save(True)
             site = get_current_site().domain
-            sign = get_md5(get_md5(settings.SECRET_KEY + str(user.id)))
+            sign = get_sha256(get_sha256(settings.SECRET_KEY + str(user.id)))
 
             if settings.DEBUG:
                 site = '127.0.0.1:8000'
@@ -147,7 +147,7 @@ def account_result(request):
     '''.format(email=user.email)
             title = '注册成功'
         else:
-            c_sign = get_md5(get_md5(settings.SECRET_KEY + str(user.id)))
+            c_sign = get_sha256(get_sha256(settings.SECRET_KEY + str(user.id)))
             sign = request.GET.get('sign')
             if sign != c_sign:
                 return HttpResponseForbidden()

blog/tests.py

@@ -1,7 +1,7 @@
 from django.test import Client, RequestFactory, TestCase
 from blog.models import Article, Category, Tag, SideBar, Links
 from django.contrib.auth import get_user_model
-from DjangoBlog.utils import get_current_site, get_md5
+from DjangoBlog.utils import get_current_site, get_sha256
 from blog.forms import BlogSearchForm
 from django.core.paginator import Paginator
 from blog.templatetags.blog_tags import load_pagination_info, load_articletags
@@ -175,7 +175,7 @@ class ArticleTest(TestCase):
             file.write(rsp.content)
         rsp = self.client.post('/upload')
         self.assertEqual(rsp.status_code, 403)
-        sign = get_md5(get_md5(settings.SECRET_KEY))
+        sign = get_sha256(get_sha256(settings.SECRET_KEY))
         with open(imagepath, 'rb') as file:
             imgfile = SimpleUploadedFile(
                 'python.png', file.read(), content_type='image/jpg')

blog/views.py

@@ -12,7 +12,7 @@ from django import forms
 from django.http import HttpResponse, HttpResponseRedirect, HttpResponseForbidden
 from django.views.decorators.csrf import csrf_exempt
 from django.contrib.auth.decorators import login_required
-from DjangoBlog.utils import cache, get_md5, get_blog_setting
+from DjangoBlog.utils import cache, get_sha256, get_blog_setting
 from django.shortcuts import get_object_or_404
 from blog.models import Article, Category, Tag, Links, LinkShowType
 from comments.forms import CommentForm
@@ -275,7 +275,7 @@ def fileupload(request):
         sign = request.GET.get('sign', None)
         if not sign:
             return HttpResponseForbidden()
-        if not sign == get_md5(get_md5(settings.SECRET_KEY)):
+        if not sign == get_sha256(get_sha256(settings.SECRET_KEY)):
             return HttpResponseForbidden()
         response = []
         for filename in request.FILES:

oauth/views.py

@@ -13,7 +13,7 @@ from django.views.generic import FormView, RedirectView
 from oauth.forms import RequireEmailForm
 from django.urls import reverse
 from django.db import transaction
-from DjangoBlog.utils import send_email, get_md5, save_user_avatar
+from DjangoBlog.utils import send_email, get_sha256, save_user_avatar
 from DjangoBlog.utils import get_current_site
 from django.core.exceptions import ObjectDoesNotExist
 from django.http import HttpResponseForbidden
@@ -127,10 +127,9 @@ def authorize(request):
 def emailconfirm(request, id, sign):
     if not sign:
         return HttpResponseForbidden()
-    if not get_md5(
-            settings.SECRET_KEY +
-            str(id) +
-            settings.SECRET_KEY).upper() == sign.upper():
+    if not get_sha256(settings.SECRET_KEY +
+                      str(id) +
+                      settings.SECRET_KEY).upper() == sign.upper():
         return HttpResponseForbidden()
     oauthuser = get_object_or_404(OAuthUser, pk=id)
     with transaction.atomic():
@@ -204,8 +203,8 @@ class RequireEmailView(FormView):
         oauthuser = get_object_or_404(OAuthUser, pk=oauthid)
         oauthuser.email = email
         oauthuser.save()
-        sign = get_md5(settings.SECRET_KEY +
-                       str(oauthuser.id) + settings.SECRET_KEY)
+        sign = get_sha256(settings.SECRET_KEY +
+                          str(oauthuser.id) + settings.SECRET_KEY)
         site = get_current_site().domain
         if settings.DEBUG:
             site = '127.0.0.1:8000'

servermanager/robot.py

@@ -21,7 +21,7 @@ from servermanager.api.blogapi import BlogApi
 from servermanager.api.commonapi import TuLing
 import os
 import json
-from DjangoBlog.utils import get_md5
+from DjangoBlog.utils import get_sha256
 from django.conf import settings
 import jsonpickle
 from servermanager.models import commands
@@ -202,7 +202,7 @@ class MessageHandler():
             passwd = settings.WXADMIN
             if settings.TESTING:
                 passwd = '123'
-            if passwd.upper() == get_md5(get_md5(info)).upper():
+            if passwd.upper() == get_sha256(get_sha256(info)).upper():
                 self.userinfo.isPasswordSet = True
                 self.savesession()
                 return "验证通过,请输入命令或者要执行的命令代码:输入helpme获得帮助"