victims stealing backup in separated module
This commit is contained in:
Your Name
parent
75f3a27cfc
commit
63789c2b09
2 changed files with 105 additions and 95 deletions
103
src/victimsbackup.py
Normal file
103
src/victimsbackup.py
Normal file
|
|
@ -0,0 +1,103 @@
|
|||
# from ubot import default_directory # Circular import error happens
|
||||
|
||||
from loguru import logger
|
||||
from telethon import events
|
||||
from datetime import datetime
|
||||
import json
|
||||
import re
|
||||
|
||||
|
||||
async def bio_backup_stealing(client, c, conn, default_directory):
|
||||
@client.on(events.NewMessage(outgoing=True, pattern=r'\.biostealbackup'))
|
||||
async def bio_steal_backup(event):
|
||||
cmd = event.text.split(' ', 1)
|
||||
if len(cmd) > 1:
|
||||
cmd = cmd[1].lower()
|
||||
if cmd == 'me':
|
||||
logger.info('Requested steal yourself backup...')
|
||||
else:
|
||||
logger.info('Stealing backup...')
|
||||
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}")
|
||||
logger.success(f'backup file saved to {file_path}')
|
||||
victims = None
|
||||
raw_victims = None
|
||||
file_format = None
|
||||
with open(file_path, 'r') as stealed_backup:
|
||||
if file_path.lower().endswith('.json'):
|
||||
victims = json.load(stealed_backup)
|
||||
file_format = 'json'
|
||||
await event.edit('Processing json victims...')
|
||||
elif file_path.lower().endswith('.txt'):
|
||||
raw_victims = stealed_backup.readlines()
|
||||
file_format = 'txt'
|
||||
await event.edit('Processing raw txt victims...')
|
||||
else:
|
||||
await event.edit('Format not supported, avalaible: txt, json')
|
||||
return
|
||||
|
||||
added = 0
|
||||
rejected = 0
|
||||
my_victims_ids = []
|
||||
if file_format == 'json':
|
||||
for v in victims:
|
||||
user_id = int(v['user_id'])
|
||||
profit = v['profit']
|
||||
when = v['from_infect']
|
||||
expr = v['until_infect']
|
||||
if cmd == 'me':
|
||||
my_victims_ids.append(user_id)
|
||||
c.execute("INSERT OR REPLACE INTO avocado(user_id,when_int,bio_str,bio_int,expr_int) VALUES (?, ?, ?, ?, ?)",
|
||||
(int(user_id), int(when), str(profit), int(profit), int(expr)))
|
||||
added += 1
|
||||
else:
|
||||
if not c.execute(f'SELECT user_id FROM avocado WHERE user_id == {user_id}').fetchone() and not c.execute(f'SELECT user_id FROM avocado_exclude WHERE user_id == {user_id}').fetchone():
|
||||
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))
|
||||
added += 1
|
||||
else:
|
||||
rejected += 1
|
||||
elif file_format == 'txt':
|
||||
when = int(datetime.timestamp(event.date))
|
||||
for raw_v in raw_victims:
|
||||
if raw_v == '':
|
||||
continue
|
||||
user_id = re.findall(r'tg://openmessage\?user_id=(\d+)', raw_v)
|
||||
if not user_id:
|
||||
continue
|
||||
user_id = int(user_id[0])
|
||||
profit = re.findall(r'([0-9\.\,k]+) опыта', raw_v)
|
||||
if not profit:
|
||||
continue
|
||||
profit = profit[0]
|
||||
if ',' in profit:
|
||||
profit = re.sub(r',', r'.', profit)
|
||||
if 'k' in profit:
|
||||
profit_int = int(
|
||||
float(re.sub('k', '', profit)) * 1000)
|
||||
else:
|
||||
profit_int = int(profit)
|
||||
if not c.execute(f'SELECT user_id FROM avocado WHERE user_id == {user_id}').fetchone() and not c.execute(f'SELECT user_id FROM avocado_exclude WHERE user_id == {user_id}').fetchone():
|
||||
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_int), 0))
|
||||
added += 1
|
||||
logger.debug(f'added {user_id} - {profit_int}')
|
||||
else:
|
||||
rejected += 1
|
||||
conn.commit()
|
||||
logger.success('backup success stealed')
|
||||
if cmd == 'me':
|
||||
my_victims_ids = tuple(my_victims_ids)
|
||||
result = c.execute(f'UPDATE avocado SET expr_int = 0 WHERE user_id NOT IN {my_victims_ids}').fetchall()
|
||||
conn.commit()
|
||||
logger.success('database rebased')
|
||||
del my_victims_ids
|
||||
del victims # free memory
|
||||
del raw_victims
|
||||
if cmd == 'me':
|
||||
rebased = len(result)
|
||||
await event.edit(f'Success added/updated {added} patients\nOther {rebased} patients reset to 0')
|
||||
del result
|
||||
else:
|
||||
await event.edit(f'Success added {added} new patients\nRejected or exists: {rejected}')
|
||||
97
ubot.py
97
ubot.py
|
|
@ -1,6 +1,6 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
# https://docs-python.ru/packages/telegram-klient-telethon-python/ <-info
|
||||
from src import avocmine
|
||||
from src import avocmine, victimsbackup
|
||||
import asyncio
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
|
|
@ -253,6 +253,7 @@ async def main():
|
|||
return response
|
||||
|
||||
####################################################################
|
||||
await victimsbackup.bio_backup_stealing(client, c, conn, default_directory)
|
||||
|
||||
@client.on(events.NewMessage(pattern='.*йобнув.*|.*подверг(ла)?.*|.*infected.*|.*сикди.*|.*насрал.*|.*выебал.*|.*за допомогою довіреності.*|.*by authorization infected.*|.*при помощи анонимуса атаковала.*'))
|
||||
@logger.catch
|
||||
|
|
@ -601,100 +602,6 @@ async def main():
|
|||
await asyncio.sleep(random.uniform(1.234, 4.222))
|
||||
await event.respond(f'биоеб {patient}')
|
||||
|
||||
@client.on(events.NewMessage(outgoing=True, pattern=r'\.biostealbackup'))
|
||||
async def bio_steal_backup(event):
|
||||
cmd = event.text.split(' ', 1)
|
||||
if len(cmd) > 1:
|
||||
cmd = cmd[1].lower()
|
||||
if cmd == 'me':
|
||||
logger.info('Requested steal yourself backup...')
|
||||
else:
|
||||
logger.info('Stealing backup...')
|
||||
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}")
|
||||
logger.success(f'backup file saved to {file_path}')
|
||||
victims = None
|
||||
raw_victims = None
|
||||
file_format = None
|
||||
with open(file_path, 'r') as stealed_backup:
|
||||
if file_path.lower().endswith('.json'):
|
||||
victims = json.load(stealed_backup)
|
||||
file_format = 'json'
|
||||
await event.edit('Processing json victims...')
|
||||
elif file_path.lower().endswith('.txt'):
|
||||
raw_victims = stealed_backup.readlines()
|
||||
file_format = 'txt'
|
||||
await event.edit('Processing raw txt victims...')
|
||||
else:
|
||||
await event.edit('Format not supported, avalaible: txt, json')
|
||||
return
|
||||
|
||||
added = 0
|
||||
rejected = 0
|
||||
my_victims_ids = []
|
||||
if file_format == 'json':
|
||||
for v in victims:
|
||||
user_id = int(v['user_id'])
|
||||
profit = v['profit']
|
||||
when = v['from_infect']
|
||||
expr = v['until_infect']
|
||||
if cmd == 'me':
|
||||
my_victims_ids.append(user_id)
|
||||
c.execute("INSERT OR REPLACE INTO avocado(user_id,when_int,bio_str,bio_int,expr_int) VALUES (?, ?, ?, ?, ?)",
|
||||
(int(user_id), int(when), str(profit), int(profit), int(expr)))
|
||||
added += 1
|
||||
else:
|
||||
if not c.execute(f'SELECT user_id FROM avocado WHERE user_id == {user_id}').fetchone() and not c.execute(f'SELECT user_id FROM avocado_exclude WHERE user_id == {user_id}').fetchone():
|
||||
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))
|
||||
added += 1
|
||||
else:
|
||||
rejected += 1
|
||||
elif file_format == 'txt':
|
||||
when = int(datetime.timestamp(event.date))
|
||||
for raw_v in raw_victims:
|
||||
if raw_v == '':
|
||||
continue
|
||||
user_id = re.findall(r'tg://openmessage\?user_id=(\d+)', raw_v)
|
||||
if not user_id:
|
||||
continue
|
||||
user_id = int(user_id[0])
|
||||
profit = re.findall(r'([0-9\.\,k]+) опыта', raw_v)
|
||||
if not profit:
|
||||
continue
|
||||
profit = profit[0]
|
||||
if ',' in profit:
|
||||
profit = re.sub(r',', r'.', profit)
|
||||
if 'k' in profit:
|
||||
profit_int = int(
|
||||
float(re.sub('k', '', profit)) * 1000)
|
||||
else:
|
||||
profit_int = int(profit)
|
||||
if not c.execute(f'SELECT user_id FROM avocado WHERE user_id == {user_id}').fetchone() and not c.execute(f'SELECT user_id FROM avocado_exclude WHERE user_id == {user_id}').fetchone():
|
||||
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_int), 0))
|
||||
added += 1
|
||||
logger.debug(f'added {user_id} - {profit_int}')
|
||||
else:
|
||||
rejected += 1
|
||||
conn.commit()
|
||||
logger.success('backup success stealed')
|
||||
if cmd == 'me':
|
||||
my_victims_ids = tuple(my_victims_ids)
|
||||
result = c.execute(f'UPDATE avocado SET expr_int = 0 WHERE user_id NOT IN {my_victims_ids}').fetchall()
|
||||
conn.commit()
|
||||
logger.success('database rebased')
|
||||
del my_victims_ids
|
||||
del victims # free memory
|
||||
del raw_victims
|
||||
if cmd == 'me':
|
||||
rebased = len(result)
|
||||
await event.edit(f'Success added/updated {added} patients\nOther {rebased} patients reset to 0')
|
||||
del result
|
||||
else:
|
||||
await event.edit(f'Success added {added} new patients\nRejected or exists: {rejected}')
|
||||
|
||||
@client.on(events.NewMessage(outgoing=True, pattern=r'\.biocheck$'))
|
||||
async def set_default_check_chat(event):
|
||||
states.where_send_check_avocado = event.peer_id
|
||||
|
|
|
|||
Loading…
Reference in a new issue