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

🩹 add alias duplication avoiding logic

This commit is contained in:
Rayan fernandes 2021-06-11 23:37:48 +05:30
parent 6799af0572
commit d067faed7b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/views/__init__.py
View File

@ -37,12 +37,14 @@ class Views(
def generate_alias_id(self, chat):
chat_id = chat.id
title = chat.title
while True:
orig_id = f"{title}{chat_id}" # the original id
alias_id = base64.urlsafe_b64encode(hashlib.md5(orig_id.encode()).digest())[:self.url_len].decode()
orig_id = f"{chat_id}" # the original id
unique_hash = hashlib.md5(orig_id.encode()).digest()
alias_id = base64.urlsafe_b64encode(unique_hash).decode()[:self.url_len]
if alias_id in self.chat_ids:
self.url_len += 1 # increment url_len just incase the hash is already used.
self.url_len += 1 # increment url_len just incase the hash is already used.
continue
elif (self.url_len > SHORT_URL_LEN): # reset url_len to initial if hash was unique.
self.url_len = SHORT_URL_LEN