main.py

import gif
import os
import pickle
import configparser
import time
import subprocess

ACCOUNTS_FILE = 'accounts.zoomgtadonotopenverysecret'
CONFIG_FILE = 'config.zoomgta.ini'
gif = gif.Gif()
cfg = configparser.ConfigParser()
current_account = 0
if os.path.exists(CONFIG_FILE) and os.path.getsize(CONFIG_FILE) > 0:
    cfg.read(CONFIG_FILE)
    try:
        def_gif = cfg['General']['DefaultGifPath']
    except:
        def_gif = None
    try:
        def_delay = cfg['General']['DefaultDelay']
    except:
        def_delay = 1
    try:
        api_key = cfg['General']['api-key']
    except:
        api_key = ''
else:
    def_gif = None
    def_delay = 1
    api_key = ''


if os.path.exists(ACCOUNTS_FILE) and os.path.getsize(ACCOUNTS_FILE) > 0:
    accounts = pickle.load(open(ACCOUNTS_FILE, 'rb'))
else:
    accounts = []
# start_gui = input('Start GUI? (y/n)')
# if start_gui == 'y':
#     import gui
#     exit(0)
# TODO: GUI
print('https://github.com/pozhiloy-enotik/zoom-gta')


def save_account(account):
    accounts.append(account)
    with open(ACCOUNTS_FILE, 'wb') as file:
        pickle.dump(accounts, file)


def delete_account(number):
    accounts.pop(number - 1)
    with open(ACCOUNTS_FILE, 'wb') as file:
        pickle.dump(accounts, file)


def save_config(dgif, ddelay):
    if def_gif:
        cfg['General']['DefaultGifPath'] = dgif
        cfg['General']['DefaultDelay'] = ddelay
    else:
        try:
            cfg.add_section('General')
        except:
            pass
        cfg.set('General', 'DefaultGifPath', dgif)
        cfg.set('General', 'DefaultDelay', ddelay)
    with open(CONFIG_FILE, 'w') as configfile:
        cfg.write(configfile)

def save_api(api):
    if api_key:
        cfg['General']['api-key'] = api
    else:
        try:
            cfg.add_section('General')
        except:
            pass
        cfg.set('General', 'api-key', api)
    with open(CONFIG_FILE, 'w') as configfile:
        cfg.write(configfile)

def get_captcha(file):
    captcha = ''
    print('Processing...')
    result = gif.process_phantomjs(api_key)
    if result == 2:
        print('Enter your phantomjscloud api-key \n'
              'Введи ключ(Читай faq):')
        key = input().strip()
        gif.process_phantomjs(key)
        save_api(key)
    try:
        if "com.termux" in os.environ.get("PREFIX", ""):  # If device is running Termux (thanks crinny)
            path = f'/sdcard/{file}'
            gif.get_captcha(path)
            subprocess.run(
                ["am", "start", "--user", "0", "-a", "android.intent.action.VIEW", "-d", f'file://{path}', "-t",
                 "image/png", "--activity-clear-task"],
                stdout=subprocess.DEVNULL,
                stderr=subprocess.DEVNULL,
            )
            captcha = input('Enter captcha(saved as "captcha.png"):\n')
    except FileNotFoundError:
        pass
    try:
        if not captcha:
            gif.get_captcha(file)
            import gui
            captcha = gui.start_gui(file)
    except:
        captcha = input(f'Enter captcha(saved as "{file}"):\n')
    return str(captcha)


def re_log_in():
    gif.make_session()
    print('\nCookies expired')
    email, password, auth_cookies = accounts[current_account]
    captcha = get_captcha('captcha.png')
    # print(captcha)
    gif.get_temp_cookies()
    result = gif.get_auth_cookies(email, password, captcha)
    gif.process_cookies()
    if result:
        delete_account(current_account + 1)
        save_account((email, password, gif.session.cookies))
        return True
    else:
        exit(0)

def log_in():
    global current_account
    i = 1
    gif.make_session()
    while i <= len(accounts):
        print(str(i) + '.', accounts[i - 1][0])
        i += 1
    choice = int(input(str(i) + '. ' + 'Add an account\n' +
                       str(i + 1) + '. ' + 'Delete an account\n'))
    if choice == i:
        email = input('Enter your e-mail:\n')
        password = input('Enter your password:\n')

        captcha = get_captcha('captcha.png')
        # print(captcha)
        gif.get_temp_cookies()
        result = gif.get_auth_cookies(email, password, captcha)

        gif.process_cookies()

        if result:
            save_account((email, password, gif.session.cookies))

            current_account = len(accounts) - 1
            return True
        else:
            exit(0)


    elif choice == i + 1:
        delete_account(int(input('Enter the number of the account\n')))
        log_in()
        return
    else:
        current_account = choice - 1
        email, password, auth_cookies = accounts[choice - 1]
        gif.session.cookies = auth_cookies
        gif.get_temp_cookies()
        gif.process_cookies()


def upload():
    if not def_gif:
        file = input('Enter the name of a .gif:\n')
    else:
        file = input('Enter the name of a .gif ( ' + def_gif + ' ) :\n')
        if not file:
            file = def_gif
    path = os.path.join(os.path.curdir, file)
    result = gif.process_image(path)
    if result == 1:
        print('Not a .gif')
        return upload()
    elif result == 2:
        re_log_in()
        return upload()
    return file


def set_delay():
    if not def_gif:
        gif.delay = float(input('Enter the delay:\n').replace(',', '.'))
    else:
        inp = input('Enter the delay ( ' + str(def_delay) + ' ) :\n')
        if inp == '':
            gif.delay = float(def_delay)
        else:
            gif.delay = float(inp.replace(',', '.'))
    return str(gif.delay)


log_in()
save_config(upload(), set_delay())

while True:
    print('Starting... Press Ctrl+C to stop')
    length = len(gif.images)
    i = 0
    try:
        while True:

            if i == length:
                i = 0
            status = gif.change_picture(0, 0, f'{gif.w}', f'{gif.h}', gif.images[i])

            if status['status']:
                print('Frame:', i+1)
                i += 1
            else:
                if status["errorMessage"] == 503 or status["errorMessage"] == 403 or status["errorMessage"] == 502:
                    print('Too many requests. Waiting for 10 seconds')
                    time.sleep(10)
                else:
                    print('Frame', i, f'skipped. An error occurred: "{status["errorMessage"]}"')
    except KeyboardInterrupt:
        print('Stopped')
        a = int(input('1. Start\n'
                      '2. Change account\n'
                      '3. Change gif\n'
                      '4. Change delay\n'
                      '5. Exit\n'))
        if a == 1:
            pass
        elif a == 2:
            log_in()
        elif a == 3:
            upload()
        elif a == 4:
            set_delay()
        elif a == 5:
            exit()