mastotuit/mastodon-setup.py

import getpass
from mastodon import Mastodon
from mastodon.Mastodon import MastodonMalformedEventError, MastodonNetworkError, MastodonReadTimeout, MastodonAPIError, MastodonIllegalArgumentError
import fileinput,re
import os
import sys

def create_dir():
  if not os.path.exists('secrets'):
    os.makedirs('secrets')

def create_file():
  if not os.path.exists('secrets/secrets.txt'):
    with open('secrets/secrets.txt', 'w'): pass
    print(secrets_filepath + " created!")

def create_config():
  if not os.path.exists('config'):
    os.makedirs('config')
  if not os.path.exists(config_filepath):
    print(config_filepath + " created!")
    with open('config/config.txt', 'w'): pass

def write_params():
  with open(secrets_filepath, 'a') as the_file:
    print("Writing secrets parameter names to " + secrets_filepath)
    the_file.write('uc_client_id: \n'+'uc_client_secret: \n'+'uc_access_token: \n')

def write_config():
  with open(config_filepath, 'a') as the_file:
    the_file.write('mastodon_hostname: \n')
    print("adding parameter name 'mastodon_hostname' to "+ config_filepath)

def read_client_lines(self):
  client_path = 'app_clientcred.txt'
  with open(client_path) as fp:
    line = fp.readline()
    cnt = 1
    while line:
       if cnt == 1:
         print("Writing client id to " + secrets_filepath)
         modify_file(secrets_filepath, "uc_client_id: ", value=line.rstrip())
       elif cnt == 2:
         print("Writing client secret to " + secrets_filepath)
         modify_file(secrets_filepath, "uc_client_secret: ", value=line.rstrip())
       line = fp.readline()
       cnt += 1

def read_token_line(self):
  token_path = 'app_usercred.txt'
  with open(token_path) as fp:
    line = fp.readline()
    print("Writing access token to " + secrets_filepath)
    modify_file(secrets_filepath, "uc_access_token: ", value=line.rstrip())

def read_config_line():
  with open(config_filepath) as fp:
    line = fp.readline()
    modify_file(config_filepath, "mastodon_hostname: ", value=hostname)

def log_in():
  error = 0
  try:
    global hostname
    hostname = input("Enter Mastodon hostname: ")
    user_name = input("User name, ex. user@" + hostname +"? ")
    user_password = getpass.getpass("User password? ")
    app_name = input("This app name? ")
    Mastodon.create_app(app_name, scopes=["read","write"],
      to_file="app_clientcred.txt", api_base_url=hostname)
    mastodon = Mastodon(client_id = "app_clientcred.txt", api_base_url = hostname)
    mastodon.log_in(
       user_name,
       user_password,
       scopes = ["read", "write"],
       to_file = "app_usercred.txt"
    )
  except MastodonIllegalArgumentError as i_error:
    error = 1
    if os.path.exists("secrets/secrets.txt"):
      print("Removing secrets/secrets.txt file..")
      os.remove("secrets/secrets.txt")
    if os.path.exists("app_clientcred.txt"):
      print("Removing app_clientcred.txt file..")
      os.remove("app_clientcred.txt")
    sys.exit(i_error)
  except MastodonNetworkError as n_error:
    error = 1
    if os.path.exists("secrets/secrets.txt"):
      print("Removing secrets/secrets.txt file..")
      os.remove("secrets/secrets.txt")
    if os.path.exists("app_clientcred.txt"):
      print("Removing app_clientcred.txt file..")
      os.remove("app_clientcred.txt")
    sys.exit(n_error)
  except MastodonReadTimeout as r_error:
    error = 1
    if os.path.exists("secrets/secrets.txt"):
      print("Removing secrets/secrets.txt file..")
      os.remove("secrets/secrets.txt")
    if os.path.exists("app_clientcred.txt"):
      print("Removing app_clientcred.txt file..")
      os.remove("app_clientcred.txt")
    sys.exit(r_error)
  except MastodonAPIError as a_error:
    error = 1
    if os.path.exists("secrets/secrets.txt"):
      print("Removing secrets/secrets.txt file..")
      os.remove("secrets/secrets.txt")
    if os.path.exists("app_clientcred.txt"):
      print("Removing app_clientcred.txt file..")
      os.remove("app_clientcred.txt")
    sys.exit(a_error)
  finally:
    if error == 0:

      create_dir()
      create_file()
      write_params()
      client_path = 'app_clientcred.txt'
      read_client_lines(client_path)
      token_path = 'app_usercred.txt'
      read_token_line(token_path)
      if os.path.exists("app_clientcred.txt"):
        print("Removing app_clientcred.txt temp file..")
        os.remove("app_clientcred.txt")
      if os.path.exists("app_usercred.txt"):
        print("Removing app_usercred.txt temp file..")
        os.remove("app_usercred.txt")
      print("Secrets setup done!\n")

def modify_file(file_name,pattern,value=""):
    fh=fileinput.input(file_name,inplace=True)
    for line in fh:
        replacement=pattern + value
        line=re.sub(pattern,replacement,line)
        sys.stdout.write(line)
    fh.close()

# Returns the parameter from the specified file
def get_parameter( parameter, file_path ):
    # Check if secrets file exists
    if not os.path.isfile(file_path):
        print("File %s not found, creating it."%file_path)
        log_in()

    # Find parameter in file
    with open( file_path ) as f:
        for line in f:
            if line.startswith( parameter ):
                return line.replace(parameter + ":", "").strip()

    # Cannot find parameter, exit
    print(file_path + " Missing parameter %s "%parameter)
    sys.exit(0)

# Returns the parameter from the specified file
def get_hostname( parameter, config_filepath ):
    # Check if secrets file exists
    if not os.path.isfile(config_filepath):
        print("File %s not found, creating it."%config_filepath)
        create_config()

    # Find parameter in file
    with open( config_filepath ) as f:
        for line in f:
            if line.startswith( parameter ):
                return line.replace(parameter + ":", "").strip()

    # Cannot find parameter, exit
    print(config_filepath + "  Missing parameter %s "%parameter)
    write_config()
    read_config_line()
    print("hostname setup done!")
    sys.exit(0)

###############################################################################
# main

if __name__ == '__main__':

    # Load secrets from secrets file
    secrets_filepath = "secrets/secrets.txt"
    uc_client_id     = get_parameter("uc_client_id",     secrets_filepath)
    uc_client_secret = get_parameter("uc_client_secret", secrets_filepath)
    uc_access_token  = get_parameter("uc_access_token",  secrets_filepath)

    # Load configuration from config file
    config_filepath = "config/config.txt"
    mastodon_hostname = get_hostname("mastodon_hostname", config_filepath) # E.g., mastodon.social