rival_regions_wrapper/app.py

"""Test module"""

import sys
import json

from rival_regions_wrapper import Client


def read_credentials(filename):
    """Read credentials from filename"""
    with open(filename) as credential_file:
        return json.load(credential_file)

def login(credentials=None):
    """Main method"""
    if credentials is None:
        credentials = {}
        credentials['username'] = input("Username: ")
        credentials['password'] = input("Password: ")
        credentials['method'] = input("Login Method: ")

    client = Client(show_window=True)
    client.login(credentials)
    print(client.var_c)

    action_dict = {
        'market': market,
        'oil_market': oil_market,
        'article': article,
        'get': get,
        'gold_exploration': gold_exploration,
        'vote_law': vote_law,
    }
    print(action_dict.keys())
    while True:
        action = input("Action: ")
        if action in action_dict:
            action_dict[action](client)
        else:
            print('action not found')

def market(client):
    """Get all market prices"""
    market_info = client.get_all_market_info()
    for i in market_info:
        print("")
        print(i.upper())
        print("#"*len(i))
        for j in market_info[i]:
            print(j.upper() + ':' + market_info[i][j])

def oil_market(client):
    """Get oil market price"""
    print(client.market_info('oil'))

def article(client):
    """Create article"""
    client.create_article('Nothing to see here', '')

def get(client):
    """Send get request from client"""
    path = input('Path: ')
    result = client.get(path)
    print(result)

def gold_exploration(client):
    """Create gold exploration law"""
    resoure = 0
    data = {
        'tmp_gov': resoure
    }
    result = client.post('parliament/donew/42/{}/0'.format(resoure), data)
    print(result)

def vote_law(client):
    """Vote for a law"""
    # p400220003260451563564814
    # p4002 2000326045 1563564814
    # 'parliament/votelaw/4002/2000326045/1563564814/pro'
    # 'parliament/votelaw/4002/2000326045/1563565114/pro'
    region_id = 4002
    player_id = 2000326045
    law_id = 1563565114
    result = client.post('parliament/votelaw/{}/{}/{}/pro'.format(
        region_id,
        player_id,
        law_id
    ), {})
    print(result)
    

if __name__ == "__main__":
    if len(sys.argv) >= 2:
        CREDENTIALS = read_credentials(sys.argv[1])
        login(CREDENTIALS)
    else:
        login()