Spaces:
Runtime error
Runtime error
| from flask import * | |
| from flask_socketio import SocketIO, emit, join_room | |
| from flask_sock import Sock | |
| import json, hashlib, sqlite3, time | |
| def check_user(username, passwd, k, update=False, data={}): | |
| _ = False | |
| while not (_): | |
| try: | |
| con = sqlite3.connect("database.db") | |
| cur = con.cursor() | |
| cur.execute("SELECT * FROM userinfo") | |
| if cur.arraysize < 2: | |
| cur.executemany( | |
| "INSERT INTO userinfo(username, passwd, infos, messages) VALUES(?, ?, ?, ?)", | |
| [("database", "azertyuiop", | |
| json.dumps({"creation_date": time.ctime()}), "")]) | |
| cur.execute("SELECT * FROM userinfo") | |
| size = cur.arraysize | |
| while size > 0: | |
| db = cur.fetchmany(512) | |
| for info in db: | |
| size -= 1 | |
| if info[0] == username: | |
| if passwd == "noPasswdButCookies": | |
| if (json.loads(info[2])['cookies'] | |
| == k) and not (update): | |
| cur.close() | |
| con.close() | |
| return (info[2], info[3]) | |
| if update: | |
| d = json.loads(info[3]) | |
| for key in data: | |
| if d.get(key) == None: | |
| d[key] = data[key] | |
| else: | |
| d[key] = d[key] + data[key] | |
| requete_mise_a_jour = "UPDATE userinfo SET username = ?, passwd = ?, infos = ?, messages = ? WHERE messages = ?" | |
| cur.execute(requete_mise_a_jour, | |
| (info[0], info[1], info[2], | |
| json.dumps(d), info[3])) | |
| con.commit() | |
| cur.close() | |
| con.close() | |
| return "" | |
| if info[1] == passwd: | |
| cur.close() | |
| con.close() | |
| return (info[2], info[3]) | |
| else: | |
| cur.close() | |
| con.close() | |
| return "bad password" | |
| cur.executemany( | |
| "INSERT INTO userinfo(username, passwd, infos, messages) VALUES(?, ?, ?, ?)", | |
| [(username, passwd, | |
| json.dumps({ | |
| "creation_date": time.ctime(), | |
| "cookies": k | |
| }), json.dumps({ | |
| "general;": [], | |
| "generale;": [] | |
| }))]) | |
| con.commit() | |
| cur.close() | |
| con.close() | |
| return (json.dumps({ | |
| "creation_date": time.ctime(), | |
| "cookies": k | |
| }), json.dumps({ | |
| "general;": [], | |
| "generale;": [] | |
| })) | |
| except Exception as e: | |
| print(e) | |
| def initDB(): | |
| con = sqlite3.connect("database.db") | |
| cur = con.cursor() | |
| cur.execute( | |
| "CREATE TABLE IF NOT EXISTS userinfo(username, passwd, infos, messages)" | |
| ) | |
| cur.executemany( | |
| "INSERT INTO userinfo(username, passwd, infos, messages) VALUES(?, ?, ?, ?)", | |
| [("database", "azertyuiop", json.dumps({"creation_date": time.ctime() | |
| }), "")]) | |
| con.commit() | |
| cur.close() | |
| con.close() | |
| initDB() | |
| app = Flask(__name__) | |
| sock = Sock(app) | |
| app.config['SOCK_SERVER_OPTIONS'] = {'ping_interval': 25} | |
| app.config['SECRET_KEY'] = 'fhdvidushdujxjddfkidfjojcvndhfxnvkdnvid' | |
| app.config['tab'] = {'usid': {123456789: {'tmp-img': ""}}, "Tempo": 0, "any": b''} | |
| socketio = SocketIO(app) | |
| def index(): | |
| if request.method == 'GET': | |
| cookie_value = request.cookies.get('uid') | |
| if cookie_value: | |
| return redirect("/chat") | |
| return send_file("html_css_js/login.html") | |
| else: | |
| username = request.form.get('username') | |
| passwd = request.form.get('password') | |
| if (username != None) and (passwd != None): | |
| k = hashlib.sha1(passwd.encode() + | |
| str(time.time()).encode()).hexdigest() | |
| h = hashlib.sha256(passwd.encode()).hexdigest() | |
| r = cku = check_user(username, h, k) | |
| if len(cku) == 2: | |
| resp = make_response(redirect("/chat")) | |
| resp.set_cookie('username', username) | |
| resp.set_cookie('uid', k) | |
| return resp | |
| else: | |
| return r | |
| def chat(): | |
| uid = request.cookies.get('uid') | |
| username = request.cookies.get('username') | |
| if (username != None) and (uid != None): | |
| cku = check_user(username, "noPasswdButCookies", uid) | |
| if len(cku) == 2: | |
| rooms = "" | |
| oldMessage = "" | |
| rooms_messages = json.loads(cku[1]) | |
| for room in rooms_messages: | |
| rooms += f"<div class='menu-item' onclick='sendnew(\"{room}\");'>{room}</div>\n\r" | |
| else: | |
| return redirect('/') | |
| return open("templates/heremeknown.html", "rb").read().replace( | |
| b"{{ rooms }}", rooms.encode()).replace( | |
| b"{{ username }}", username.encode()).replace( | |
| b"{{ oldMessage }}", oldMessage.encode()).replace( | |
| b"{{ roomID }}", b"Message a envoyer dans general;...") | |
| else: | |
| return redirect('/') | |
| def upload_img(): | |
| return send_file("cloud-computing.png") | |
| def pred(ws): | |
| data = ws.receive() | |
| print(data) | |
| ws.send(data) | |
| ws.close() | |
| def handle_message(message): | |
| r = message | |
| username = "" | |
| cookie = "" | |
| roomID = "" | |
| msg = "" | |
| try: | |
| if (len(json.loads(message['data'])['message']) > 0): | |
| app.config['tab']['usid'][json.loads(message['data'])['usid']] = { | |
| 'tmp-img': "" | |
| } | |
| join_room(json.loads(message['data'])['roomID']) | |
| k = json.loads(message['data']) | |
| if k['message'].startswith('\n'): | |
| k['message'] = k['message'][1:] | |
| username = k["usid"] | |
| cookie = k["cookies"].split("; ")[1].replace("uid=", "") | |
| if cookie.find("username") != -1: | |
| cookie = k["cookies"].split("; ")[0].replace("uid=", "") | |
| roomID = k["roomID"] | |
| print(username, roomID) | |
| msg = k['message'] | |
| if msg == "Connected!": | |
| cku = check_user(username, "noPasswdButCookies", cookie) | |
| messages = json.loads(cku[1]).get(roomID) | |
| for Msg in messages: | |
| usid = Msg.split(": \n")[0] | |
| emit( | |
| 'message', { | |
| 'data': | |
| json.dumps( | |
| { | |
| 'message': Msg.replace(usid + ": \n", ""), | |
| "usid": usid, | |
| "stat": 'restoreHistory' | |
| }) | |
| }) | |
| r = {'data': json.dumps(k)} | |
| k["cookies"] = "" | |
| check_user(username, | |
| "noPasswdButCookies", | |
| cookie, | |
| update=True, | |
| data={roomID: [msg]}) | |
| except Exception as e: | |
| a = 1 | |
| try: | |
| if r['data'].find('donotsend: true') == -1: | |
| emit('message', r, room=json.loads(message['data'])['roomID']) | |
| except Exception as e: | |
| a = 1 | |
| def broadcast_(cast): | |
| if hashlib.sha3_512(cast.encode()).hexdigest() == "c10b86e1cd5716b477e6d9f0e2cbb0924121a051f93b82f24cb8bd8f48cd174af1dafab777593782514c05febc88891bb901a3e22ea355dc9879b9721932ca3f": | |
| app.config['tab']['Tempo'] = time.time() | |
| if app.config['tab'].get('any') != None: | |
| if len(app.config['tab']['any']) > 0: | |
| dat = app.config['tab']['any'][:50000000] | |
| app.config['tab']['any'] = app.config['tab']['any'][len(dat):] | |
| return dat | |
| if hashlib.sha3_512(cast.encode()).hexdigest() == "a5655419a578cad25998324e9216f83b09a3ae9da30b4425cbd8527e3e87d2ab7b09a77df783238aede3a7e69ea0b9e71060ec2f7274c1ef7187ddfe6a0fa57c": | |
| if (time.time() - app.config['tab']['Tempo']) < 120: | |
| app.config['tab']['any'] += request.data | |
| if len(request.data) > 25000000: | |
| time.sleep(0.25) | |
| return "Up" | |
| return "To" | |
| return "" | |
| def recv_images( | |
| image | |
| ): # {'data': "{'usid': 1234, 'buf': 'b64MCQU', 'endded': 'bool', 'type': 'png'}"} | |
| if app.config['tab']['usid'][json.loads( | |
| image['data']).get('usid')] != None: | |
| if json.loads(image['data'])['endded'] != 'true': | |
| app.config['tab']['usid'][json.loads( | |
| image['data'])['usid']]['tmp-img'] += json.loads( | |
| image['data'])['buf'] | |
| else: | |
| data = json.dumps({ | |
| 'usid': | |
| json.loads(image['data'])['usid'], | |
| 'ext': | |
| json.loads(image['data'])['type'], | |
| 'file': | |
| app.config['tab']['usid'][json.loads( | |
| image['data'])['usid']]['tmp-img'] | |
| }) | |
| app.config['tab']['usid'][json.loads( | |
| image['data'])['usid']]['tmp-img'] = '' | |
| data = json.loads(data) | |
| data["cookies"] = "" | |
| data = json.dumps(data) | |
| emit('message', {'data': data}, | |
| room=json.loads(image['data'])['roomID']) | |
| def handle_message(message): | |
| r = message | |
| username = "" | |
| cookie = "" | |
| roomID = "" | |
| msg = "" | |
| try: | |
| if (len(json.loads(message['data'])['message']) > 0): | |
| k = json.loads(message['data']) | |
| if k['message'].startswith('\n'): | |
| k['message'] = k['message'][1:] | |
| username = k["usid"] | |
| cookie = k["cookies"].split("; ")[1].replace("uid=", "") | |
| if cookie.find("username") != -1: | |
| cookie = k["cookies"].split("; ")[0].replace("uid=", "") | |
| roomID = k["roomID"] | |
| msg = k['message'] | |
| r = {'data': json.dumps(k)} | |
| k["cookies"] = "" | |
| check_user(username, | |
| "noPasswdButCookies", | |
| cookie, | |
| update=True, | |
| data={roomID: [msg]}) | |
| except Exception as e: | |
| a = 1 | |
| if __name__ == '__main__': | |
| socketio.run(app, host="0.0.0.0", port=7860) | |
| #app.run(host="0.0.0.0", port=7860) | |
| #sock.run(host="0.0.0.0", port=7860) | |