79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
|
import webbrowser
|
||
|
|
|
||
|
|
import click
|
||
|
|
import pendulum
|
||
|
|
from mastodon import Mastodon
|
||
|
|
|
||
|
|
from config import (
|
||
|
|
USER_SECRET_FILE,
|
||
|
|
APP_SECRET_FILE,
|
||
|
|
API_BASE_URL,
|
||
|
|
PROTECTED_TOOTS,
|
||
|
|
BOOST_LIMIT,
|
||
|
|
FAVORITE_LIMIT,
|
||
|
|
DAYS_COUNT,
|
||
|
|
TIMEZONE,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@click.group()
|
||
|
|
def cli():
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@cli.command()
|
||
|
|
def create_app():
|
||
|
|
Mastodon.create_app(
|
||
|
|
"cleantoots", api_base_url=API_BASE_URL, to_file=APP_SECRET_FILE,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@cli.command()
|
||
|
|
def get_credentials():
|
||
|
|
mastodon = Mastodon(client_id=APP_SECRET_FILE)
|
||
|
|
webbrowser.open(mastodon.auth_request_url())
|
||
|
|
code = click.prompt("Enter code")
|
||
|
|
mastodon.log_in(
|
||
|
|
code=code, to_file=USER_SECRET_FILE,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@cli.command()
|
||
|
|
@click.option("--delete", is_flag=True)
|
||
|
|
def clean_toots(delete):
|
||
|
|
mastodon = Mastodon(access_token=USER_SECRET_FILE)
|
||
|
|
user = mastodon.me()
|
||
|
|
page = mastodon.account_statuses(user["id"])
|
||
|
|
would_delete = []
|
||
|
|
while page:
|
||
|
|
for toot in page:
|
||
|
|
if (
|
||
|
|
toot["reblogs_count"] >= BOOST_LIMIT
|
||
|
|
or toot["favourites_count"] >= FAVORITE_LIMIT
|
||
|
|
or toot["id"] in PROTECTED_TOOTS
|
||
|
|
or toot["created_at"]
|
||
|
|
>= pendulum.now(tz=TIMEZONE).subtract(days=DAYS_COUNT)
|
||
|
|
):
|
||
|
|
continue
|
||
|
|
would_delete.append(toot)
|
||
|
|
|
||
|
|
page = mastodon.fetch_next(page)
|
||
|
|
|
||
|
|
if not delete:
|
||
|
|
click.secho(
|
||
|
|
"Would delete {count} toots:".format(count=len(would_delete)), fg="blue"
|
||
|
|
)
|
||
|
|
for toot in would_delete:
|
||
|
|
click.echo(toot["id"])
|
||
|
|
click.echo(toot["content"])
|
||
|
|
click.echo()
|
||
|
|
else:
|
||
|
|
click.echo("Deleting toots...")
|
||
|
|
with click.progressbar(would_delete) as bar:
|
||
|
|
for toot in bar:
|
||
|
|
mastodon.status_delete(toot)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
cli()
|