Sakuten/backend

#!/usr/bin/env python3
import sys
import os
sys.path.append(os.getcwd())  # noqa: E402
from cards.id import load_id_json_file
from urllib.request import Request, urlopen
from urllib.error import HTTPError
from pathlib import Path
import json


class client():
    def get(self, _url, _json=None, follow_redirects=False, headers=None):
        default_headers = {"Content-Type": "application/json"}
        if headers:
            default_headers.update(headers)
        json_data = json.dumps(_json).encode("utf-8") if _json else None

        url = f'http://localhost:8888{_url}'
        request = Request(url, data=json_data,
                          headers=default_headers, method='GET')
        try:
            response = urlopen(request)
        except HTTPError as e:
            print(e.read().decode(), file=sys.stderr)
            sys.exit(-1)
        else:
            response_body = response.read().decode("utf-8")
            response.close()
            return json.loads(response_body)

    def post(self, _url, _json=None, follow_redirects=False, headers=None):
        default_headers = {"Content-Type": "application/json"}
        if headers:
            default_headers.update(headers)
        json_data = json.dumps(_json).encode("utf-8") if _json else None
        url = f'http://localhost:8888{_url}'
        request = Request(url, data=json_data,
                          headers=default_headers, method='POST')

        try:
            response = urlopen(request)
        except HTTPError as e:
            print(e.read().decode(), file=sys.stderr)
            sys.exit(-1)
        else:
            response_body = response.read().decode("utf-8")
            response.close()
            return json.loads(response_body)


def login(client, secret_id, rresp):
    return client.post('/auth', _json={
        "id": secret_id,
        "g-recaptcha-response": rresp
    }, follow_redirects=True)


client = client()


id_list = load_id_json_file(Path(__file__).parent.parent /
                            Path('cards/test_users.json'))
users = [i for i in id_list if i['authority'] == 'normal']

lotteries = client.get('/lotteries/available')

if not lotteries:
    print('no lottery available')
    sys.exit(69)

else:
    n_users_per_lot = len(users) // len(lotteries)

    for i, lottery in enumerate(lotteries):
        print(f'applying to {lottery["id"]}: ', end='')
        for user in users[n_users_per_lot * i:n_users_per_lot * (i + 1) - 1]:
            token = login(client, user['secret_id'], '')['token']
            client.post(f'/lotteries/{lottery["id"]}',
                        _json={"group_members": ""},
                        headers={"Authorization": f"Bearer {token}"})
            print('.', end='')
        print(' DONE')

    print('all lotteries are treated')