import requests
import base64
import json
import os
import time
import string
import random
import names

egelauth="Basic XXXX"
#egelcompanyid="XXXXX" #dev
#egelcompanyid="XXXX"
egelaccountid="XXX" #stable
i=0
password="XXXX"
urlbasis="http://dev.alpha-trader.com/"
partnerid="XXXXX" #stable
#partnerid="XXXXX"
#partnerid="XXXXX"#dev
headeregel={
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
auth=""

def createuser(username):
	print("createuser")
	print("\n")
	url = urlbasis+"user/register"
	mail=username+"@gmx.de"
	companyname="comp"+username
	querystring = {"username":username,"emailAddress":mail,"password":password}
	headers = {
		'x-authorization': partnerid,
    	'cache-control': "no-cache",
  	 	}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")

def createcompany(companyname, username):
	print("createcompany")
	print("\n")
	url = urlbasis+"api/companies/"
	querystring = {"name":companyname,"cashDeposit":"50000"}
	headers = {
   	 'authorization': auth,
   	 'x-authorization': partnerid,
   	 'cache-control': "no-cache",
   		}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")

def getcompanyidbyuser(username):
	print("getcompanyidbyuser")
	print("\n")
	url = urlbasis+"api/companies/ceo/username/"+username
	response = requests.request("GET", url, headers=headeregel)
	parsed_json = json.loads(response.text)
	companyid=parsed_json[0]['id']
	return companyid

def employegelasceo(companyid):
	print("employegelasceo")
	print("\n")
	url = urlbasis+"api/polls/employceo"
	querystring = {"companyId":companyid,"dailyWage":"150"}
	response = requests.request("POST", url, headers=headeregel, params=querystring)
	print(response.text)
	print("\n")
	url = urlbasis+"api/polls/"
	headers = {
    	'authorization': auth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}

	response = requests.request("GET", url, headers=headers)
	parsed_json = json.loads(response.text)
	pollid=parsed_json[0]['id']
	print(pollid)

	url = urlbasis+"api/polls/"+pollid

	querystring = {"voices":"50000","votingType":"YES"}
	headers = {
    	'authorization': auth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")
	#print(response.text)

	url = urlbasis+"api/polls/execute/"+pollid
	response = requests.request("POST", url, headers=headeregel)

def getsecurityidentifier(companyname):
	print("getsecurityidentifier")
	print("\n")
	url = urlbasis+"api/search/companies/"+companyname

	headers = {
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
	response = requests.request("GET", url, headers=headers)
	print(response.text)
	print("\n")
	parsed_json = json.loads(response.text)
	securityidentifier=parsed_json[0]['securityIdentifier']
	return securityidentifier

def getsecurityaccount(companyname):
	print("getsecurityaccount")
	print("\n")
	url = urlbasis+"api/search/companies/"+companyname

	headers = {
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
	response = requests.request("GET", url, headers=headers)
	print(response.text)
	print("\n")
	parsed_json = json.loads(response.text)
	securityaccount=parsed_json[0]['securitiesAccountId']
	return securityaccount

def getcompanyidbyname(companyname):
	print("getcompanyidbyname")
	print("\n")
	url = urlbasis+"api/search/companies/"+companyname

	headers = {
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
	response = requests.request("GET", url, headers=headers)
	print(response.text)
	print("\n")
	parsed_json = json.loads(response.text)
	securityaccount=parsed_json[0]['id']
	return securityaccount

def otcordersell(company1, company2):
	print("otcordersell")
	print("\n")
	url = urlbasis+"api/securityorders/"
	querystring = {"owner":getsecurityaccount(company1),"securityIdentifier":getsecurityidentifier(company1),"action":"SELL","type":"LIMIT","price":"1","numberOfShares":"50000","counterparty":getsecurityaccount(company2)}
	headers = {
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",

    	}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")
	querystring = {"owner":getsecurityaccount(company2),"securityIdentifier":getsecurityidentifier(company1),"action":"BUY","type":"LIMIT","price":"1","numberOfShares":"50000","counterparty":getsecurityaccount(company1)}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")

def intitiateliquidationpoll(company):
	print("initiateliquidationpoll")
	print("\n")
	url = urlbasis+"api/polls/liquidation"
	querystring = {"companyId":getcompanyidbyname(company)}
	headers = {
    	'authorization': egelauth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",
    	}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")

#def acceptliquidationpoll(company)
def sellowncompany(company, auth):
	print("sellowncompany")
	print("\n")
	url = urlbasis+"api/securityorders/"
	querystring = {"owner":getsecurityaccount(company),"securityIdentifier":getsecurityidentifier(company),"action":"SELL","type":"LIMIT","price":"1","numberOfShares":"50000","counterparty":""}
	headers = {
    	'authorization': auth,
    	'x-authorization': partnerid,
    	'cache-control': "no-cache",

    	}
	response = requests.request("POST", url, headers=headers, params=querystring)
	print(response.text)
	print("\n")

def getlastpoll():
	print("getlastpoll")
	print("\n")
	url = urlbasis+"api/initiatedpolls/"
	response = requests.request("GET", url, headers=headeregel)

	parsed_json = json.loads(response.text)
	#print(parsed_json.lentgh())
	pollid=parsed_json[-1]['id']
	return pollid

def votepoll (pollid):
	print("votepoll")
	print("\n")
	url = urlbasis+"api/polls/"+pollid
	querystring = {"voices":"50000","votingType":"YES"}
	response = requests.request("POST", url, headers=headeregel, params=querystring)


def expoll (pollid):
	print("createuser")
	print("\n")
	url = urlbasis+"api/polls/execute/"+pollid
	response = requests.request("POST", url, headers=headeregel)

def id_generator(size=60, chars=string.ascii_uppercase + string.digits):
	return ''.join(random.choice(chars) for _ in range(size))

def randomName(capitalize):

    bits=[]
    vowels="aeiou"
    letters="abcdefghijklmnopqrstuvwxyz"
    for ch in letters:
        for v in vowels:
            bits.append(ch+v)
    bits.remove("fu")
    bits.remove("hi")
    bits.remove("cu")
    bits.remove("co")
    bits.remove("mo")
    word=""
    rnd=len(bits)-1
    numOfBits=random.randint(2,3)
    for i in range(0,numOfBits):
        word=word+bits[random.randint(1,rnd)]
    word=word+letters[random.randrange(0,25)]
    if (capitalize==True):
        word=word.capitalize()
    return word


i=1
while i<100:
	username="ZZZ"+id_generator()
	#username=names.get_first_name()
	ag=username
	auth=username+":"+password
	auth= base64.encodestring(auth)
	auth="Basic "+auth
	auth=auth.replace('\n', '')
	createuser(username)
	createcompany(ag, username)
	print(username)
	#employegelasceo(getcompanyidbyuser(username))
	otcordersell(ag, "Bender")
	#time.sleep(2)
	#intitiateliquidationpoll(ag)
	#lastpoll=getlastpoll()
	#votepoll(lastpoll)
	#expoll(getlastpoll())
	i=i+1
	print(username)




#employegelasceo(getcompanyidbyuser(username))
#auth=egelauth
#comp="Bis%20zum%20naechsten%20reset"
#otcordersell(comp, "Bender")
#intitiateliquidationpoll(comp)
#getlastpoll()
#print(getcompanyidbyname(comp))