forust/userbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore(userbot): apply ruff check --fix and ruff format

Browse Source 
- ruff check --fix: 210 auto-fixed errors (import sorting, trailing
  whitespace, unused imports, f-string fixups, deprecated annotations)
- ruff format: 104 files reformatted to consistent style
- 268 non-auto-fixable issues remain (S113 requests timeout, etc.)
This commit is contained in:
mr-forust
2026-06-19 12:19:01 +02:00
parent 95cec59263
commit 7ba6bc44f2
104 changed files with 4338 additions and 5319 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/rentry.py
+49 -62
View File
@@ -5,18 +5,17 @@ import asyncio
import http.cookiejar
import urllib.parse
import urllib.request
from uuid import uuid4
from datetime import datetime, timedelta
from http.cookies import SimpleCookie
from json import loads as json_loads
from uuid import uuid4
from utils.db import db
BASE_PROTOCOL = "https://"
BASE_URL = "rentry.co"
BASE_PROTOCOL = 'https://'
BASE_URL = 'rentry.co'
_headers = {"Referer": f"{BASE_PROTOCOL}{BASE_URL}"}
_headers = {'Referer': f'{BASE_PROTOCOL}{BASE_URL}'}
class UrllibClient:
@@ -24,9 +23,7 @@ class UrllibClient:
def __init__(self):
self.cookie_jar = http.cookiejar.CookieJar()
self.opener = urllib.request.build_opener(
urllib.request.HTTPCookieProcessor(self.cookie_jar)
)
self.opener = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(self.cookie_jar))
urllib.request.install_opener(self.opener)
def get(self, url, headers=None):
@@ -45,58 +42,50 @@ class UrllibClient:
def _request(self, request):
response = self.opener.open(request)
response.status_code = response.getcode()
response.data = response.read().decode("utf-8")
response.data = response.read().decode('utf-8')
return response
def raw(url: str):
client = UrllibClient()
return json_loads(client.get(f"{BASE_PROTOCOL}{BASE_URL}/api/raw/{url}").data)
return json_loads(client.get(f'{BASE_PROTOCOL}{BASE_URL}/api/raw/{url}').data)
def new(text: str, edit_code: str = "", url: str = ""):
def new(text: str, edit_code: str = '', url: str = ''):
client, cookie = UrllibClient(), SimpleCookie()
cookie.load(vars(client.get(f"{BASE_PROTOCOL}{BASE_URL}"))["headers"]["Set-Cookie"])
csrftoken = cookie["csrftoken"].value
cookie.load(vars(client.get(f'{BASE_PROTOCOL}{BASE_URL}'))['headers']['Set-Cookie'])
csrftoken = cookie['csrftoken'].value
payload = {
"csrfmiddlewaretoken": csrftoken,
"url": url,
"edit_code": edit_code,
"text": text,
'csrfmiddlewaretoken': csrftoken,
'url': url,
'edit_code': edit_code,
'text': text,
}
return json_loads(
client.post(
f"{BASE_PROTOCOL}{BASE_URL}" + "/api/new", payload, headers=_headers
).data
)
return json_loads(client.post(f'{BASE_PROTOCOL}{BASE_URL}' + '/api/new', payload, headers=_headers).data)
def edit(url_short: str, edit_code: str, text: str):
client, cookie = UrllibClient(), SimpleCookie()
cookie.load(vars(client.get(f"{BASE_PROTOCOL}{BASE_URL}"))["headers"]["Set-Cookie"])
csrftoken = cookie["csrftoken"].value
cookie.load(vars(client.get(f'{BASE_PROTOCOL}{BASE_URL}'))['headers']['Set-Cookie'])
csrftoken = cookie['csrftoken'].value
payload = {"csrfmiddlewaretoken": csrftoken, "edit_code": edit_code, "text": text}
payload = {'csrfmiddlewaretoken': csrftoken, 'edit_code': edit_code, 'text': text}
return json_loads(
client.post(
f"{BASE_PROTOCOL}{BASE_URL}/api/edit/{url_short}", payload, headers=_headers
).data
)
return json_loads(client.post(f'{BASE_PROTOCOL}{BASE_URL}/api/edit/{url_short}', payload, headers=_headers).data)
def delete(url_short: str, edit_code: str):
client, cookie = UrllibClient(), SimpleCookie()
cookie.load(vars(client.get(f"{BASE_PROTOCOL}{BASE_URL}"))["headers"]["Set-Cookie"])
csrftoken = cookie["csrftoken"].value
payload = {"csrfmiddlewaretoken": csrftoken, "edit_code": edit_code}
cookie.load(vars(client.get(f'{BASE_PROTOCOL}{BASE_URL}'))['headers']['Set-Cookie'])
csrftoken = cookie['csrftoken'].value
payload = {'csrfmiddlewaretoken': csrftoken, 'edit_code': edit_code}
return json_loads(
client.post(
f"{BASE_PROTOCOL}{BASE_URL}/api/delete/{url_short}",
f'{BASE_PROTOCOL}{BASE_URL}/api/delete/{url_short}',
payload,
headers=_headers,
).data
@@ -128,35 +117,35 @@ async def paste(
if edit_bin:
if not (url and edit_code):
raise ValueError("Please provide both, url and edit code")
raise ValueError('Please provide both, url and edit code')
response = edit(url_short=url, edit_code=edit_code, text=text)
else:
response = new(text=text)
if response.get("status") != "200":
if response.get('status') != '200':
raise RuntimeError(
f"paste task terminated with status: {response.get('status')}\n"
f"Message: {response.get('content', 'No message provided')}"
f'paste task terminated with status: {response.get("status")}\n'
f'Message: {response.get("content", "No message provided")}'
)
url = response["url"]
edit_code = response["edit_code"]
url = response['url']
edit_code = response['edit_code']
if not permanent:
short_url = response["url_short"]
short_url = response['url_short']
time_now = datetime.now()
ftime = time_now.strftime("%d %I:%M:%S %p %Y")
ftime = time_now.strftime('%d %I:%M:%S %p %Y')
print(f"URL: {url} - Edit Code: {edit_code} - Time: {ftime}")
print(f'URL: {url} - Edit Code: {edit_code} - Time: {ftime}')
rallUrls = db.get("core.rentry", "urls", default={"allUrls": {}})
rallUrls = db.get('core.rentry', 'urls', default={'allUrls': {}})
entry_id = str(uuid4())
rallUrls["allUrls"][entry_id] = {
"url": short_url,
"edit_code": edit_code,
"time": ftime,
rallUrls['allUrls'][entry_id] = {
'url': short_url,
'edit_code': edit_code,
'time': ftime,
}
db.set("core.rentry", "urls", rallUrls)
db.set('core.rentry', 'urls', rallUrls)
if return_edit:
return (url, edit_code)
@@ -167,34 +156,32 @@ async def rentry_cleanup_job():
"""Periodically checks and deletes rentry pastes older than 24 hours"""
while True:
try:
rallUrls = db.get("core.rentry", "urls", default={"allUrls": {}})
rallUrls = db.get('core.rentry', 'urls', default={'allUrls': {}})
now = datetime.now()
deleted_count = 0
error_count = 0
for entry_id, entry in list(rallUrls["allUrls"].items()):
url = entry["url"]
entry_time = datetime.strptime(entry["time"], "%d %I:%M:%S %p %Y")
for entry_id, entry in list(rallUrls['allUrls'].items()):
url = entry['url']
entry_time = datetime.strptime(entry['time'], '%d %I:%M:%S %p %Y')
if now - entry_time > timedelta(days=1):
try:
delete(url, entry["edit_code"])
del rallUrls["allUrls"][entry_id]
delete(url, entry['edit_code'])
del rallUrls['allUrls'][entry_id]
deleted_count += 1
print(f"[#] Deleted expired rentry paste: {url}")
print(f'[#] Deleted expired rentry paste: {url}')
except Exception as e:
error_count += 1
print(f"[!] Failed to delete rentry paste {url}: {str(e)}")
print(f'[!] Failed to delete rentry paste {url}: {str(e)}')
if deleted_count or error_count:
print(
f"[*] Cleanup summary: {deleted_count} deleted, {error_count} failed"
)
print(f'[*] Cleanup summary: {deleted_count} deleted, {error_count} failed')
if deleted_count:
db.set("core.rentry", "urls", rallUrls)
db.set('core.rentry', 'urls', rallUrls)
except Exception as e:
print(f"[!] Error in rentry cleanup job: {str(e)}")
print(f'[!] Error in rentry cleanup job: {str(e)}')
await asyncio.sleep(12 * 60 * 60)