Ver código fonte

버그 수정

2DU 8 anos atrás
pai
commit
50e0c07ffc
2 arquivos alterados com 13 adições e 4 exclusões
  1. 5 2
      app.py
  2. 8 2
      func.py

+ 5 - 2
app.py

@@ -3593,11 +3593,14 @@ def upload():
             return(re_error('/error/9'))
 
         if(int(wiki_set(3)) * 1024 * 1024 < request.content_length):
-            return re_error('/error/17')
+            return(re_error('/error/17'))
         
         value = os.path.splitext(data.filename)[1]
+        if(not value):
+            return(re_error('/error/16'))
+
         if(not value in ['.jpeg', '.jpg', '.gif', '.png', '.webp', '.JPEG', '.JPG', '.GIF', '.PNG', '.WEBP']):
-            return re_error('/error/14')
+            return(re_error('/error/14'))
     
         if(request.forms.get('f_name')):
             name = request.forms.get('f_name') + value

+ 8 - 2
func.py

@@ -2,7 +2,7 @@
 from bottle.ext import beaker
 import json
 import sqlite3
-from hashlib import md5
+import hashlib
 from urllib import parse
 import re
 import html
@@ -28,6 +28,9 @@ def get_time():
     date = "%04d-%02d-%02d %02d:%02d:%02d" % (now.tm_year, now.tm_mon, now.tm_mday, now.tm_hour, now.tm_min, now.tm_sec)
 
     return(date)
+
+def sha224(data):
+    return(hashlib.sha224(bytes(data, 'utf-8')).hexdigest())
     
 def ip_check():
     session = request.environ.get('beaker.session')
@@ -171,7 +174,7 @@ def ip_pas(raw_ip):
         curs.execute("select data from other where name = 'ip_view'")
         d = curs.fetchall()
         if(d and d[0][0] != ''):
-            ip = '<span style="font-size: 75%;">' + md5(bytes(raw_ip, 'utf-8')).hexdigest() + '</span>'
+            ip = '<span style="font-size: 75%;">' + hashlib.md5(bytes(raw_ip, 'utf-8')).hexdigest() + '</span>'
             if(not admin_check('ban', None)):
                 hide = 1
         else:
@@ -445,6 +448,9 @@ def re_error(data):
         elif(num == 12):
             title = '편집 오류'
             data = '누군가 먼저 편집 했습니다.'
+        elif(num == 16):
+            title = '파일 올리기 오류'
+            data = '파일 이름을 다른 걸로 설정 해주세요.'
         elif(num == 14):
             title = '파일 올리기 오류'
             data = 'jpg, gif, jpeg, png, webp만 가능 합니다.'