Przeglądaj źródła

비 로그인 커스텀 CSS 지원

2DU 9 lat temu
rodzic
commit
e1e27da756
2 zmienionych plików z 19 dodań i 16 usunięć
  1. 16 10
      app.py
  2. 3 6
      func.py

+ 16 - 10
app.py

@@ -2071,7 +2071,6 @@ def logout():
     session = request.environ.get('beaker.session')
     session['Now'] = False
     session.pop('DREAMER', None)
-    session.pop('Daydream', None)
 
     return redirect('/user')
     
@@ -2541,12 +2540,11 @@ def user_info():
 @route('/custom', method=['GET', 'POST'])
 def custom_css():
     session = request.environ.get('beaker.session')
-    if(not session.get('Now') == True):
-        return redirect('/login')
-    else:
-        ip = ip_check()
 
-        if(request.method == 'POST'):
+    ip = ip_check()
+
+    if(request.method == 'POST'):
+        if(not re.search('\.', ip)):
             db_ex("select * from custom where user = '" + db_pas(ip) + "'")
             css_data = db_get()
             if(css_data):
@@ -2555,18 +2553,26 @@ def custom_css():
                 db_ex("insert into custom (user, css) value ('" + db_pas(ip) + "', '" + db_pas(request.forms.content) + "')")
             db_com()
 
-            session['Daydream'] = request.forms.content
+        session['Daydream'] = request.forms.content
 
-            return redirect('/user')
-        else:
+        return redirect('/user')
+    else:
+        if(not re.search('\.', ip)):
+            start = ''
             db_ex("select * from custom where user = '" + db_pas(ip) + "'")
             css_data = db_get()
             if(css_data):
                 data = css_data[0]['css']
             else:
                 data = ''
+        else:
+            start = '<span>비 로그인의 경우에는 로그인하면 날아갑니다.</span><br><br>'
+            try:
+                data = session['Daydream']
+            except:
+                data = ''
 
-            return template('other', custom = custom_css_user(), license = set_data['license'], login = login_check(), title = '커스텀 CSS', logo = set_data['name'], data = '<form id="usrform" name="f1" method="POST" action="/custom"><textarea rows="30" cols="100" name="content" form="usrform">' + data + '</textarea><div class="form-actions"><button class="btn btn-primary" type="submit">저장</button></div></form>')
+        return template('other', custom = custom_css_user(), license = set_data['license'], login = login_check(), title = '커스텀 CSS', logo = set_data['name'], data = start + '<form id="usrform" name="f1" method="POST" action="/custom"><textarea rows="30" cols="100" name="content" form="usrform">' + data + '</textarea><div class="form-actions"><button class="btn btn-primary" type="submit">저장</button></div></form>')
     
 @route('/count')
 def count_edit():

+ 3 - 6
func.py

@@ -186,12 +186,9 @@ def ip_check():
 
 def custom_css_user():
     session = request.environ.get('beaker.session')
-    if(session.get('Now') == True):
-        try:
-            data = format(session['Daydream'])
-        except:
-            data = ''
-    else:
+    try:
+        data = format(session['Daydream'])
+    except:
         data = ''
 
     return data