EngineTribe/common.py at main · KeiDongo/EngineTribe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import base64
from dataclasses import dataclass
from enum import Enum
import hashlib
import re

from xpinyin import Pinyin

from locales import *

from database.models import Level

from models import LevelDetails, LevelDetailsUserData


class ClientType(Enum):
    STABLE = 1
    TESTING = 2
    LEGACY = 3
    ENGINE_BOT = 4


@dataclass
class AuthCodeData:
    user_id: int
    platform: str
    locale: str
    locale_item: LocaleModel
    legacy: bool
    testing_client: bool


def level_to_details(level_data: Level, locale: str, level_file_url: str, mobile: bool, like_type: str,
                     clear_type: str,
                     author: str, record_user: str):
    if mobile and level_data.non_latin:
        name: str = string_latinify(level_data.name)
    else:
        name: str = level_data.name
    if level_data.record != 0:
        record = {'record': 'yes',
                  'alias': record_user,
                  'id': level_data.record_user_id,
                  'time': level_data.record}
    else:
        record = {'record': 'no'}

    desc = level_data.description
    if desc == '' or desc is None:
        desc = 'Sin descripción'

    return LevelDetails(
        name=name,
        likes=level_data.likes,
        dislikes=level_data.dislikes,
        comments=0,
        intentos=level_data.plays,
        muertes=level_data.deaths,
        victorias=level_data.clears,
        apariencia=level_data.style,
        entorno=level_data.environment,
        etiquetas=f'{prettify_tag_name(level_data.tag_1, locale)},{prettify_tag_name(level_data.tag_2, locale)}',
        featured=int(level_data.featured),
        user_data=LevelDetailsUserData(
            completed=clear_type,
            liked=like_type
        ),
        date=level_data.date.strftime("%m/%d/%Y"),
        author=author,
        record=record,
        archivo=level_file_url,
        id=level_data.level_id,
        descripcion=desc,
    )


def gen_level_id_md5(stripped_swe: str) -> str:
    return prettify_level_id(hashlib.md5(stripped_swe.encode()).hexdigest().upper()[8:24])


def gen_level_id_sha1(stripped_swe: str) -> str:
    return prettify_level_id(hashlib.sha1(stripped_swe.encode()).hexdigest().upper()[8:24])


def gen_level_id_sha256(stripped_swe: str) -> str:
    return prettify_level_id(hashlib.sha256(stripped_swe.encode()).hexdigest().upper()[8:24])


def strip_level(data_swe: str) -> str:
    result = base64.b64decode(data_swe)[:-30].decode("UTF-8")
    regex_time = re.compile('"time": ".*?"')
    regex_date = re.compile('"date": ".*?"')
    result = regex_date.sub('"date": ""', regex_time.sub('"time": ""', result))
    return result


def prettify_level_id(level_id: str):
    return level_id[0:4] + '-' + level_id[4:8] + '-' + level_id[8:12] + '-' + level_id[12:16]


def calculate_password_hash(password: str):
    return hashlib.sha256(base64.b64encode(password.encode('utf-8'))).hexdigest()


def string_latinify(t):
    table = {ord(f): ord(t) for f, t in zip(u'，。！？【】（）％＃＠＆－—〔〕：；〇﹒—﹙﹚、—“”', u',.!?[]()%#@&--():;0.—(),-""')}

    try:
        t2 = t.translate(table)
    except:
        t2 = t
    t2 = Pinyin().get_pinyin(t2).replace('-', ' ')
    t2 = re.sub(u'[^\x00-\x7F\x80-\xFF\u0100-\u017F\u0180-\u024F\u1E00-\u1EFF]', u'', t2)
    return t2