Class GithubPlugin

pokemongo_bot/plugin_loader.py:48–149

    return getattr(my_module, class_name)

class GithubPlugin(object):
  PLUGINS_FOLDER = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'plugins')

  def __init__(self, plugin_name):
    self.plugin_name = plugin_name
    self.plugin_parts = self.get_github_parts()

  def is_valid_plugin(self):
    return self.plugin_parts is not None

  def get_github_parts(self):
    groups = re.match('(.*)\/(.*)#(.*)', self.plugin_name)

    if groups is None:
      return None

    parts = {}
    parts['user'] = groups.group(1)
    parts['repo'] = groups.group(2)
    parts['sha'] = groups.group(3)

    return parts

  def get_installed_version(self):
    if not self.is_already_installed():
      return None

    filename = os.path.join(self.get_plugin_folder(), '.sha')
    print(filename)
    with open(filename) as file:
        return file.read().strip()

  def get_local_destination(self):
    parts = self.plugin_parts
    if parts is None:
      raise Exception('Not a valid github plugin')

    file_name = '{}_{}_{}.zip'.format(parts['user'], parts['repo'], parts['sha'])
    full_path = os.path.join(self.PLUGINS_FOLDER, file_name)
    return full_path

  def is_already_installed(self):
    file_path = self.get_plugin_folder()
    if not os.path.isdir(file_path):
      return False

    sha_file = os.path.join(file_path, '.sha')

    if not os.path.isfile(sha_file):
      return False

    with open(sha_file) as file:
      content = file.read().strip()

      if content != self.plugin_parts['sha']:
        return False

    return True