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(credentials, show_window=True)
    print(client.var_c)

    action = input("Action: ")
    action_dict = {
        "market": market,
        "oil_market": oil_market,
        "article": article
    }

    if action in action_dict:
        action_dict[action](client)

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("Test", "Whoops")

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