seedminer_autolauncher.py

#!/usr/bin/env python3

import os
import requests
import os.path
import getpass
import sys
import signal
import time
import re
import glob

s = requests.Session()
baseurl = "https://seedhelper.figgyc.uk"
currentid = ""

# https://stackoverflow.com/a/16696317 thx
def download_file(url, local_filename):
    # NOTE the stream=True parameter
    r = requests.get(url, stream=True)
    with open(local_filename, 'wb') as f:
        for chunk in r.iter_content(chunk_size=1024):
            if chunk: # filter out keep-alive new chunks
                f.write(chunk)
                #f.flush() commented by recommendation from J.F.Sebastian
    return local_filename


print("Updating seedminer db...")
    os.system('"' + sys.executable + '" seedminer_launcher3.py update_db')

#username = input("Username: ")
#password = getpass.getpass("Password: ")
#print("Logging in...")
#r = s.post(baseurl + "/login", data={'username': username, 'password': password})
#print(r.url)
#if r.url == baseurl + '/home':
#    print("Login successful")
#else:
#    print("Login fail")
#    sys.exit(1)

def signal_handler(signal, frame):
        print('Exiting...')
        s.get(baseurl + "/cancel/" + currentid)
        sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)



while True:
    print("Finding work...")
    r = s.get(baseurl + "/getwork")
    if r.text == "nothing":
        print("No work. Waiting 30 seconds...")
        time.sleep(30)
    else:
        currentid = r.text
        r2 = s.get(baseurl + "/claim/" + currentid)
        if r2.text == "error":
            print("Device already claimed, trying again...")
        else:
            print("Downloading part1 for device " + currentid)
            download_file(baseurl + '/part1/' + currentid, 'movable_part1.sed')
            print("Bruteforcing")
            os.system('"' + sys.executable + '" seedminer_launcher3.py gpu')
            if os.path.isfile("movable.sed"):
                print("Uploading")
                # seedhelper2 has no msed database but we upload these anyway if zoogie wants them or smth idk
                list_of_files = glob.glob('msed_data_*.bin') # * means all if need specific format then *.csv
                latest_file = max(list_of_files, key=os.path.getctime)
                ur = s.post(baseurl + '/upload/' + currentid, files={'movable': open('movable.sed', 'rb'), 'msed': open(latest_file, 'rb')})
                if ur.text == "good":
                    print("Upload succeeded!")
                    os.remove("movable.sed")
                    os.remove(latest_file)
                else:
                    print("Upload failed!")
                    sys.exit(1)