octospacc
/
TelegramIndex-Fork
mirror of https://gitlab.com/octospacc/TelegramIndex-Fork.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

🐛 improved hash generating logic

This commit is contained in:
Rayan fernandes 2021-06-10 14:25:44 +05:30
parent 78236dea8d
commit 18702721c6
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/views/__init__.py
View File

@ -31,24 +31,21 @@ class Views(
): ):
def __init__(self, client): def __init__(self, client):
self.client = client self.client = client
self.url_len = SHORT_URL_LEN
self.chat_ids = {} self.chat_ids = {}
def generate_alias_id(self, chat): def generate_alias_id(self, chat):
chat_id = chat.id chat_id = chat.id
title = chat.title title = chat.title
while True: while True:
# alias_id = "".join(
# [
# random.choice(string.ascii_letters + string.digits)
# for _ in range(len(str(chat_id)))
# ]
# )
orig_id = f"{title}{chat_id}" # the original id orig_id = f"{title}{chat_id}" # the original id
alias_id = base64.urlsafe_b64encode(hashlib.md5(orig_id.encode()).digest())[:SHORT_URL_LEN].decode() alias_id = base64.urlsafe_b64encode(hashlib.md5(orig_id.encode()).digest())[:self.url_len].decode()
if alias_id in self.chat_ids: if alias_id in self.chat_ids:
self.url_len += 1 # increment url_len just incase the hash is already used.
continue continue
elif (self.url_len > SHORT_URL_LEN): # reset url_len to initial if hash was unique.
self.url_len = SHORT_URL_LEN
self.chat_ids[alias_id] = { self.chat_ids[alias_id] = {
"chat_id": chat_id, "chat_id": chat_id,