openNAMU__openNAMU/route/tool/func_tool.py

import urllib.parse
import datetime
import hashlib
import flask
import re

import os
import html
import json
import sqlite3
import threading

set_data = ''

def get_time():
    return str(datetime.datetime.today().strftime("%Y-%m-%d %H:%M:%S"))

def db_data_get(data):
    global set_data
    
    set_data = data

def db_change(data):
    if set_data == 'mysql':
        data = data.replace('random()', 'rand()')
        data = data.replace('%', '%%')
        data = data.replace('?', '%s')

    return data

def ip_check(d_type = 0):
    ip = ''
    if d_type == 0 and (flask.session and 'id' in flask.session):
        ip = flask.session['id']
    else:    
        for for_a in range(5):
            if for_a == 0:
                ip = flask.request.environ.get('HTTP_X_REAL_IP', '::1')
            elif for_a == 1:
                ip = flask.request.environ.get('HTTP_CLIENT_IP', '::1')
            elif for_a == 2:
                ip = flask.request.environ.get('HTTP_X_FORWARDED_FOR', '::1')
            elif for_a == 3:
                ip = flask.request.environ.get('HTTP_CF_CONNECTING_IP', '::1')
            else:
                ip = flask.request.environ.get('REMOTE_ADDR', '::1')
            
            if type(ip) == type([]):
                ip = ip[0]
            else:
                ip = ip.split(',')
                ip = ip[0]

            if ip != '::1':
                break

    return ip

def url_pas(data):
    return urllib.parse.quote(data).replace('/','%2F')

def sha224_replace(data):
    return hashlib.sha224(bytes(data, 'utf-8')).hexdigest()

def md5_replace(data):
    return hashlib.md5(data.encode()).hexdigest()