feat: like undoing module
This commit is contained in:
69
skywipe/likes.py
Normal file
69
skywipe/likes.py
Normal file
@@ -0,0 +1,69 @@
|
||||
"""Like undoing module for Skywipe"""
|
||||
|
||||
import time
|
||||
from atproto import models
|
||||
from skywipe.auth import Auth
|
||||
from skywipe.configure import Configuration
|
||||
|
||||
|
||||
LIKE_COLLECTION = "app.bsky.feed.like"
|
||||
|
||||
|
||||
def undo_likes():
|
||||
auth = Auth()
|
||||
client = auth.login()
|
||||
config = Configuration()
|
||||
config_data = config.load()
|
||||
|
||||
batch_size = config_data.get("batch_size", 10)
|
||||
delay = config_data.get("delay", 1)
|
||||
verbose = config_data.get("verbose", False)
|
||||
|
||||
if verbose:
|
||||
print(
|
||||
f"Starting like deletion with batch_size={batch_size}, delay={delay}s")
|
||||
|
||||
did = client.me.did
|
||||
cursor = None
|
||||
total_undone = 0
|
||||
|
||||
while True:
|
||||
list_params = models.ComAtprotoRepoListRecords.Params(
|
||||
repo=did,
|
||||
collection=LIKE_COLLECTION,
|
||||
limit=batch_size,
|
||||
cursor=cursor
|
||||
)
|
||||
|
||||
response = client.com.atproto.repo.list_records(params=list_params)
|
||||
|
||||
records = response.records
|
||||
if not records:
|
||||
break
|
||||
|
||||
for record in records:
|
||||
try:
|
||||
record_uri = record.uri
|
||||
rkey = record_uri.rsplit("/", 1)[-1]
|
||||
delete_data = {
|
||||
"repo": did,
|
||||
"collection": LIKE_COLLECTION,
|
||||
"rkey": rkey
|
||||
}
|
||||
client.com.atproto.repo.delete_record(data=delete_data)
|
||||
total_undone += 1
|
||||
if verbose:
|
||||
print(f"Undone like: {record_uri}")
|
||||
except Exception as e:
|
||||
record_uri = getattr(record, "uri", "unknown")
|
||||
if verbose:
|
||||
print(f"Error undoing like {record_uri}: {e}")
|
||||
|
||||
cursor = response.cursor
|
||||
if not cursor:
|
||||
break
|
||||
|
||||
if delay > 0:
|
||||
time.sleep(delay)
|
||||
|
||||
print(f"Undone {total_undone} likes.")
|
||||
Reference in New Issue
Block a user