ephemetoot-eliminare-vecchi.../ephemetoot/console.py

172 lines
5.3 KiB
Python

#!/usr/bin/env python3
# #####################################################################
# Ephemetoot - A command line tool to delete your old toots
# Copyright (C) 2018 Hugh Rundle, 2019-2020 Hugh Rundle & others
# Initial work based on tweet-deleting script by @flesueur
# (https://gist.github.com/flesueur/bcb2d9185b64c5191915d860ad19f23f)
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# You can contact Hugh on Mastodon: @hugh@ausglam.space
# or email: ephemetoot@hugh.run
# #####################################################################
# import
import yaml
# from standard library
from argparse import ArgumentParser
from datetime import datetime, timezone
import os
import pkg_resources
# import funtions
from ephemetoot import ephemetoot as func
# version number from package info
vnum = pkg_resources.require("ephemetoot")[0].version
parser = ArgumentParser()
parser.add_argument(
"--archive-deleted",
action="store_true",
help="Only archive toots that are being deleted",
)
parser.add_argument(
"--config",
action="store",
metavar="filepath",
default="config.yaml",
help="Filepath of your config file, absolute or relative to the current directory. If no --config path is provided, ephemetoot will use 'config.yaml'in the current directory",
)
parser.add_argument(
"--datestamp",
action="store_true",
help="Include a datetime stamp for every action (e.g. deleting a toot)",
)
parser.add_argument(
"--hide-skipped",
"--hide_skipped",
action="store_true",
help="Do not write to log when skipping saved toots",
)
parser.add_argument(
"--init",
action="store_true",
help="Create a config file that is saved in the current directory",
)
parser.add_argument(
"--pace",
action="store_true",
help="Slow deletion actions to match API rate limit to avoid pausing",
)
parser.add_argument("--quiet", action="store_true", help="Suppress most logging")
parser.add_argument(
"--retry-mins",
action="store",
metavar="minutes",
nargs="?",
const="1",
default="1",
type=int,
help="Number of minutes to wait between retries, when an error is thrown",
)
parser.add_argument(
"--schedule",
action="store",
metavar="filepath",
nargs="?",
const=".",
help="Save and load plist file on MacOS",
)
parser.add_argument(
"--test", action="store_true", help="Do a test run without deleting any toots"
)
parser.add_argument(
"--time",
action="store",
metavar=("hour", "minute"),
nargs="*",
help="Hour and minute to schedule: e.g. 9 30 for 9.30am",
)
parser.add_argument(
"--verbose",
action="store_true",
help="Log more information about errors and exceptions",
)
parser.add_argument(
"--version",
action="store_true",
help="Display the version numbers of the installed and latest versions",
)
options = parser.parse_args()
if options.config[0] == "~":
config_file = os.path.expanduser(options.config)
elif options.config[0] == "/":
# make sure user isn't passing in something dodgy
if os.path.exists(options.config):
config_file = options.config
else:
config_file = ""
else:
config_file = os.path.join(os.getcwd(), options.config)
def main():
'''
Call ephemetoot.check_toots() on each user in the config file, with options set via flags from command line.
'''
try:
if options.init:
func.init()
elif options.version:
func.version(vnum)
elif options.schedule:
func.schedule(options)
else:
if not options.quiet:
print("")
print("============= EPHEMETOOT v" + vnum + " ================")
print(
"Running at "
+ str(
datetime.now(timezone.utc).strftime("%a %d %b %Y %H:%M:%S %z")
)
)
print("================================================")
print("")
if options.test:
print("This is a test run...\n")
with open(config_file) as config:
for accounts in yaml.safe_load_all(config):
for user in accounts:
func.check_toots(user, options)
except FileNotFoundError as err:
if err.filename == config_file:
print("🕵️ Missing config file")
print("Run \033[92mephemetoot --init\033[0m to create a new one\n")
else:
print("\n🤷‍♂️ The archive directory in your config file does not exist")
print("Create the directory or correct your config before trying again\n")
if __name__ == "__main__":
main()