Moved to seperate repository
This commit is contained in:
parent
30e2e1807a
commit
24747be4ee
@ -1,156 +0,0 @@
|
|||||||
#! /usr/bin/env python3
|
|
||||||
|
|
||||||
# Yandere Lewd Bot, an image posting bot for Pleroma
|
|
||||||
# Copyright (C) 2022 Anon <yanderefedi@proton.me>
|
|
||||||
#
|
|
||||||
# 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 <https://www.gnu.org/licenses/>.
|
|
||||||
|
|
||||||
import getpass
|
|
||||||
import re
|
|
||||||
import sys
|
|
||||||
import datetime
|
|
||||||
import importlib
|
|
||||||
from mastodon import Mastodon, MastodonIllegalArgumentError, MastodonAPIError, MastodonIOError
|
|
||||||
from collections import OrderedDict
|
|
||||||
import yanlib
|
|
||||||
|
|
||||||
|
|
||||||
class CreateAppError(Exception):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def get_client_id_and_secret(app, permissions, domain):
|
|
||||||
try:
|
|
||||||
return Mastodon.create_app(app, scopes=permissions, api_base_url=domain)
|
|
||||||
except MastodonIOError:
|
|
||||||
raise CreateAppError("An error occurred. Make sure the domain name is correct.")
|
|
||||||
|
|
||||||
|
|
||||||
def get_api(client_id, client_secret, domain):
|
|
||||||
api = Mastodon(client_id, client_secret, api_base_url=domain)
|
|
||||||
return api
|
|
||||||
|
|
||||||
|
|
||||||
def get_token(api, email, password, permissions):
|
|
||||||
try:
|
|
||||||
token = api.log_in(email, password, scopes=permissions)
|
|
||||||
return token
|
|
||||||
except MastodonIllegalArgumentError:
|
|
||||||
raise CreateAppError("Username or Password is incorrect.")
|
|
||||||
except MastodonAPIError:
|
|
||||||
raise CreateAppError("Could not grant scopes:", ", ".join(permissions))
|
|
||||||
|
|
||||||
|
|
||||||
def get_cfg(cfg_name):
|
|
||||||
try:
|
|
||||||
cfg = importlib.import_module(cfg_name)
|
|
||||||
return cfg
|
|
||||||
except ImportError:
|
|
||||||
raise CreateAppError("Cannot import module:", cfg_name, "Make sure you omitted the .py extension and try again")
|
|
||||||
|
|
||||||
|
|
||||||
def package_settings(app, domain, client_id, client_secret, token):
|
|
||||||
settings_server = OrderedDict([
|
|
||||||
("app_name", app),
|
|
||||||
("api_base_url", domain),
|
|
||||||
("client_id", client_id),
|
|
||||||
("client_secret", client_secret),
|
|
||||||
("access_token", token)
|
|
||||||
])
|
|
||||||
return settings_server
|
|
||||||
|
|
||||||
|
|
||||||
def get_setting_reminder(fmt):
|
|
||||||
dt_now = datetime.datetime.now()
|
|
||||||
dt_now = dt_now.replace(year=dt_now.year + 1)
|
|
||||||
settings_reminder = dt_now.strftime(fmt)
|
|
||||||
return settings_reminder
|
|
||||||
|
|
||||||
|
|
||||||
def main():
|
|
||||||
# Create App
|
|
||||||
print("Generate and register Mastodon App")
|
|
||||||
print("You can just hit enter to accept the default for prompts that have them")
|
|
||||||
print("Ctrl+C to Quit\n")
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Get instance information
|
|
||||||
app = input("Enter your app name (Default: app): ") or "app"
|
|
||||||
domain = input("URL of Instance (Default: https://yandere.cc): ") or "https://yandere.cc"
|
|
||||||
email = input("Enter your email: ")
|
|
||||||
password = getpass.getpass("Enter password: ")
|
|
||||||
print("Scopes: read, write, follow, push")
|
|
||||||
print("Separate each scope with a comma (example above).")
|
|
||||||
print("!!! Accept the default unless you intend to modify Yandere Lewd Bot !!!")
|
|
||||||
ans = input("Scopes (Default: write): ") or "write"
|
|
||||||
ans = re.sub(r"\s+", "", ans, flags=re.UNICODE)
|
|
||||||
permissions = ans.split(",")
|
|
||||||
print("Granting:", permissions)
|
|
||||||
|
|
||||||
# Begin logging in
|
|
||||||
client_id, client_secret = get_client_id_and_secret(app, permissions, domain)
|
|
||||||
api = get_api(client_id, client_secret, domain)
|
|
||||||
token = get_token(api, email, password, permissions)
|
|
||||||
|
|
||||||
# Get user setting for settings_time["long_date_format"]
|
|
||||||
# Needed for setting_reminder
|
|
||||||
print("What is the name of your main configuration file? Exclude the '.py' extension.")
|
|
||||||
ans = input("(Default: cfg): ") or "cfg"
|
|
||||||
cfg = get_cfg(ans)
|
|
||||||
|
|
||||||
# Credentials (unencrypted)
|
|
||||||
encrypt, salt = False, ""
|
|
||||||
settings_server = package_settings(app, domain, client_id, client_secret, token)
|
|
||||||
reminder = get_setting_reminder(cfg.settings_time["long_date_format"])
|
|
||||||
|
|
||||||
# Encrypt
|
|
||||||
ans = input("Do you want to encrypt your credentials (y/N)? ")
|
|
||||||
if ans.upper() in ("Y", "YES"):
|
|
||||||
import encryption
|
|
||||||
encrypt = True
|
|
||||||
salt, settings_server = encryption.settings_server_encrypt_cfg(settings_server)
|
|
||||||
|
|
||||||
settings_encrypt = OrderedDict([
|
|
||||||
("encrypt", encrypt),
|
|
||||||
("salt", salt),
|
|
||||||
])
|
|
||||||
|
|
||||||
# Output the user's new credentials
|
|
||||||
print("Copy to your config file!!!")
|
|
||||||
yanlib.pp_ordered_dict("settings_server", settings_server)
|
|
||||||
print('\nsettings_reminder = "{}"\n'.format(reminder))
|
|
||||||
yanlib.pp_dict("settings_encrypt", settings_encrypt)
|
|
||||||
|
|
||||||
return 0
|
|
||||||
except (KeyboardInterrupt, EOFError):
|
|
||||||
print("Quitting...")
|
|
||||||
return 1
|
|
||||||
except CreateAppError as e:
|
|
||||||
print(e)
|
|
||||||
return 2
|
|
||||||
except Exception as e:
|
|
||||||
print("Unhandled Exception!!", e)
|
|
||||||
return 3
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
status_code = main()
|
|
||||||
status_msg = {
|
|
||||||
0: "Success :)",
|
|
||||||
1: "User Quit :|",
|
|
||||||
2: "Error :(",
|
|
||||||
3: "Unhandled Exception ¯\\_(ツ)_/¯"
|
|
||||||
}
|
|
||||||
print(status_msg[status_code])
|
|
||||||
sys.exit(status_code)
|
|
||||||
Reference in New Issue
Block a user