73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import configparser
|
|
import os
|
|
import sys
|
|
import webbrowser
|
|
|
|
import click
|
|
|
|
|
|
def _is_tty():
|
|
return sys.stdout.isatty() and sys.stdin.isatty()
|
|
|
|
|
|
def _config_has_sections(config):
|
|
if not config.sections():
|
|
click.secho("The config file doesn't seem to have any section.", fg="yellow")
|
|
command = click.style("cleantoots config setup", bold=True)
|
|
click.secho(f"You should set it up first. Use: {command}")
|
|
return False
|
|
return True
|
|
|
|
|
|
def _open_url(url, echo):
|
|
opened = False
|
|
|
|
if _is_tty():
|
|
if echo:
|
|
click.echo(
|
|
"We will now try to open a browser for each account set in the config file."
|
|
)
|
|
click.echo(
|
|
"You'll need to authenticate and then copy the code provided in the web "
|
|
"page back into this terminal, upon prompt."
|
|
)
|
|
click.pause()
|
|
result = webbrowser.open(url)
|
|
opened = result
|
|
|
|
if not opened:
|
|
click.echo(f"Go to {url}, authenticate and enter the code below.")
|
|
|
|
|
|
def _get_default_config():
|
|
default_config = configparser.ConfigParser()
|
|
default_config["DEFAULT"] = {
|
|
"boost_limit": 5,
|
|
"favorite_limit": 5,
|
|
"days_count": 30,
|
|
"timezone": "Europe/Paris",
|
|
}
|
|
default_config["Mastodon.social"] = {
|
|
"api_base_url": "https://mastodon.social",
|
|
"app_secret_file": "mastodon_social_app.secret",
|
|
"user_secret_file": "mastodon_social_user.secret",
|
|
"protected_toots": "1234\n5678",
|
|
"protected_tags": "TagToProtect",
|
|
}
|
|
return default_config
|
|
|
|
|
|
class CleanTootsConfig(configparser.ConfigParser):
|
|
def __init__(self, config_dir, config_file_name, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.dir = config_dir
|
|
os.makedirs(self.dir, exist_ok=True)
|
|
self.main_file = os.path.join(config_dir, config_file_name)
|
|
self.read(self.main_file)
|
|
|
|
def file(self, filename):
|
|
return os.path.join(self.dir, filename)
|
|
|
|
def isfile(self, filename):
|
|
return os.path.isfile(self.file(filename))
|