S1S13AF7/ub4tg
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

more information when stealing victims backup

Browse source
This commit is contained in:
Your Name 2024-12-14 14:18:20 +03:00
parent 7635615d5a
commit 55b3586ba7
1 changed files with 9 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
ubot.py
View file

@ -599,6 +599,7 @@ async def main():
else: else:
logger.info('Stealing backup...') logger.info('Stealing backup...')
reply = await client.get_messages(event.peer_id, ids=event.reply_to.reply_to_msg_id) reply = await client.get_messages(event.peer_id, ids=event.reply_to.reply_to_msg_id)
await event.edit('Downloading file...')
file_path = await reply.download_media(file=f"{default_directory}") file_path = await reply.download_media(file=f"{default_directory}")
logger.success(f'backup file saved to {file_path}') logger.success(f'backup file saved to {file_path}')
victims = None victims = None
@ -608,14 +609,17 @@ async def main():
if file_path.lower().endswith('.json'): if file_path.lower().endswith('.json'):
victims = json.load(stealed_backup) victims = json.load(stealed_backup)
file_format = 'json' file_format = 'json'
await event.edit('Processing json victims...')
elif file_path.lower().endswith('.txt'): elif file_path.lower().endswith('.txt'):
raw_victims = stealed_backup.readlines() raw_victims = stealed_backup.readlines()
file_format = 'txt' file_format = 'txt'
await event.edit('Processing raw txt victims...')
else: else:
await event.edit('Format not supported, avalaible: txt, json') await event.edit('Format not supported, avalaible: txt, json')
return return
added = 0 added = 0
rejected = 0
my_victims_ids = [] my_victims_ids = []
if file_format == 'json': if file_format == 'json':
for v in victims: for v in victims:
@ -633,6 +637,8 @@ async def main():
c.execute("INSERT INTO avocado(user_id,when_int,bio_str,bio_int,expr_int) VALUES (?, ?, ?, ?, ?)", c.execute("INSERT INTO avocado(user_id,when_int,bio_str,bio_int,expr_int) VALUES (?, ?, ?, ?, ?)",
(int(user_id), int(when), str(profit), int(profit), 0)) (int(user_id), int(when), str(profit), int(profit), 0))
added += 1 added += 1
else:
rejected += 1
elif file_format == 'txt': elif file_format == 'txt':
when = int(datetime.timestamp(event.date)) when = int(datetime.timestamp(event.date))
for raw_v in raw_victims: for raw_v in raw_victims:
@ -658,6 +664,8 @@ async def main():
(int(user_id), int(when), str(profit), int(profit_int), 0)) (int(user_id), int(when), str(profit), int(profit_int), 0))
added += 1 added += 1
logger.debug(f'added {user_id} - {profit_int}') logger.debug(f'added {user_id} - {profit_int}')
else:
rejected += 1
conn.commit() conn.commit()
logger.success('backup success stealed') logger.success('backup success stealed')
if cmd == 'me': if cmd == 'me':
@ -673,7 +681,7 @@ async def main():
await event.edit(f'Success added/updated {added} patients\nOther {rebased} patients reset to 0') await event.edit(f'Success added/updated {added} patients\nOther {rebased} patients reset to 0')
del result del result
else: else:
await event.edit(f'Success added {added} new patients') await event.edit(f'Success added {added} new patients\nRejected or exists: {rejected}')
@client.on(events.NewMessage(outgoing=True, pattern=r'\.biocheck$')) @client.on(events.NewMessage(outgoing=True, pattern=r'\.biocheck$'))
async def set_default_check_chat(event): async def set_default_check_chat(event):