#!/usr/bin/env python #Sheychen 2017 #CC BY import sys import random import time import os.path import json import APILeekwars as API from CommandTree import CommandTree global lwapi lwapi = API.APILeekwars() class Settings: def __init__(self, options): self.filePath = options['path'] if self.filePath.startswith('*/'): self.filePath = os.path.join( os.path.dirname(__file__), self.filePath[2:]) self.settings = None self.get() def save(self): with open(self.filePath, 'w') as outfile: json.dump(self.settings, outfile) def get(self): if self.settings is None: self.settings = {'farmers': {}, 'pools': {'main': {}}} if os.path.exists( self.filePath) and os.path.getsize(self.filePath) > 0: with open(self.filePath) as json_data_file: js = json.load(json_data_file) self.settings = js return self.settings def getFarmers(self): return self.get()['farmers'] def addFarmer(self, id, login, password): if not self.settings['farmers'].get(str(id)) is None: raise ValueError('Farmer {0}: Allready added'.format(login)) self.settings['farmers'][id] = {'login': login, 'password': password} self.save() def getPools(self): return self.get()['pools'] def addPool(self, pool): if not self.settings['pools'].get(pool) is None: raise ValueError('Pool {0}: Allready added'.format(pool)) self.settings['pools'][pool] = {} self.save() def addLeek(self, pool, id, farmer): if self.settings['pools'].get(pool) is None: raise ValueError('Pool {0}: Doesn\'t exists'.format(pool)) if not self.settings['pools'][pool].get(str(id)) is None: raise ValueError('Leek {0}: Allready added in Pool {1}'.format( id, pool)) self.settings['pools'][pool][id] = str(farmer) self.save() class Farmers: def login(login, password): r = lwapi.farmer.login_token(login, password) if not r.get('success', False): raise ValueError('{0}: {1}'.format(login, r.get('error', 'Fail'))) return Farmer(r) def farmer(settings, id): login = settings.getFarmers().get(id) if login is None: raise ValueError('Can\'t find Farmer "{0}"'.format(id)) return Farmers.login(login['login'], login['password']) def get(settings): farmers = [] logins = settings.getFarmers() for id in logins: try: farmers.append( Farmers.login(logins[id]['login'], logins[id]['password'])) except ValueError as err: print(format(err)) return farmers def list(params, options): mode = params[0] if mode is None: farmers = Settings(options).getFarmers() for farmer in farmers: print('{0}: {1}'.format(farmer, farmers[farmer]['login'])) else: for farmer in Farmers.get(Settings(options)): print('{0}:'.format(farmer.name)) if mode == 'infos': for field in ['id', 'talent', 'fights', 'habs']: print(' {0} : {1}'.format(field, farmer.data[field])) elif mode == 'ais': for ai in farmer.getAis(): print(' {0} : {1}'.format(ai['name'], ai['id'])) elif mode == 'stuff': print(' {0}'.format(', '.join( str(item['template']) for item in farmer.weapons + farmer.chips))) elif mode == 'leeks': for id in farmer.leeks: print(' {0} : {1} ({2}, {3})'.format( id, farmer.leeks[id]['name'], farmer.leeks[id]['level'], farmer.leeks[id]['talent'])) def stats(params, options): print('Deprecated: use "pool stats"') mode = params[0] if mode == 'infos': fields = {'talent': [], 'fights': [], 'habs': []} for farmer in Farmers.get(Settings(options)): for field in fields: fields[field].append(farmer.data[field]) print('value : min, avg, max') for field in fields: print('{0} : {1}, {2}, {3}'.format(field, min( fields[field]), int(sum(fields[field]) / len(fields[field])), max(fields[field]))) def register(params, options): login = params[0] password = params[1] try: farmer = Farmers.login(login, password) Settings(options).addFarmer(farmer.id, login, password) farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) def buy(params, options): item = params[0] for farmer in Farmers.get(Settings(options)): try: for x in (farmer.chips + farmer.weapons): if x['template'] == item: farmer.raiseError('Allready have one') farmer.buy(item) farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) def sell(params, options): item = params[0] for farmer in Farmers.get(Settings(options)): try: farmer.sell(item) farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) class Pools: def list(params, options): print('TODO add params (leeks, ...)') print(', '.join(Settings(options).getPools())) class Pool: def parse(options): pool = options.get('pool', 'main') print('Using "{0}" pool'.format(pool)) return pool def get(settings, pid): pool = settings.getPools().get(pid) farmers = settings.getFarmers() if pool is None: raise ValueError('Pool {0}: Doesn\'t exists'.format(pid)) leeks = [] for leek in pool: try: leeks.append( Farmers.login( farmers[pool[leek]]['login'], farmers[pool[leek]]['password']).getLeek(leek)) except ValueError as err: print(format(err)) return leeks def create(params, options): try: Settings(options).addPool(Pool.parse(options)) except ValueError as err: print(format(err)) def register(params, options): try: pool = Pool.parse(options) lid = params[0] fid = None for farmer in Farmers.get(Settings(options)): if lid in farmer.leeks: fid = farmer.id if fid is None: raise ValueError( 'Any register farmer for leek "{0}"'.format(lid)) Settings(options).addLeek(pool, lid, fid) print('OK') except ValueError as err: print(format(err)) def list(params, options): try: for leek in Pool.get(Settings(options), Pool.parse(options)): print(leek.name) if len(params) == 1: mode = params[0] if mode == 'infos': for field in ['id', 'talent', 'level']: print(' {0} : {1}'.format(field, leek.data[field])) print(' fights : {0}'.format(leek.farmer.fights)) elif mode == 'stuff': for item in leek.weapons + leek.chips: print(' {0}'.format(item['template'])) elif mode == 'characteristics': for field in [ 'life', 'strength', 'wisdom', 'agility', 'resistance', 'science', 'magic', 'tp', 'mp', 'frequency', 'capital' ]: print(' {0} : {1}'.format(field, leek.data[field])) except ValueError as err: print(format(err)) def stats(params, options): mode = params[0] if mode == 'infos': fields = {'talent': [], 'level': []} elif mode == 'farmers': fields = {'fights': [], 'habs': []} elif mode == 'characteristics': fields = { 'life': [], 'strength': [], 'wisdom': [], 'agility': [], 'resistance': [], 'science': [], 'magic': [], 'tp': [], 'mp': [], 'frequency': [], 'capital': [] } try: for leek in Pool.get(Settings(options), Pool.parse(options)): for field in fields: fields[field].append(leek.data[field] if mode != 'farmers' else leek.farmer.data[field]) except ValueError as err: print(format(err)) print('value : min, avg, max') for field in fields: print('{0} : {1}, {2}, {3}'.format(field, min( fields[field]), int(sum(fields[field]) / len(fields[field])), max(fields[field]))) def fight(params, options): try: random.seed() leeks = Pool.get(Settings(options), Pool.parse(options)) leeksids = [x.id for x in leeks] random.shuffle(leeks) force = (params[1] == 'force') for leek in leeks: try: fights = params[0] if type(params[0]) is int else leek.farmer.fights if fights < 1: leek.raiseError('No more fights') for _ in range(fights): opponents = leek.getOpponents() if len(opponents) < 1: leek.raiseError('Probably, no more fights') opid = None opscore = 20000000 for x in opponents: if force or not x['id'] in leeksids: score = x['level'] * x['talent'] if score < opscore: opid = x['id'] opscore = score if opid is None: leek.raiseError( 'Really? All your opponnents are allies') print('https://leekwars.com/fight/{0}'.format( leek.fight(opid))) time.sleep(options['sleep']) except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def setupAI(params, options): try: pool = Pool.parse(options) path = os.path.join('ai', pool) if not os.path.exists(path): raise ValueError('Can\'t find "{0}" folder'.format(path)) ais = {} for fileName in os.listdir(path): if fileName.endswith('.leek'): with open(os.path.join(path, fileName), 'r') as myfile: ais[pool + '-' + fileName[:-5]] = myfile.read() if (ais.get(pool + '-AI') is None): raise ValueError('Can\'t find "{0}/AI.leek" file'.format(path)) for leek in Pool.get(Settings(options), pool): try: fais = leek.farmer.getAis() for ai in ais: try: aid = None for current in fais: if current['name'] == ai: aid = current['id'] break if aid is None: print('New ai "{0}" for {1}'.format( ai, leek.farmer.name)) aid = leek.farmer.newAi(0, ai)['id'] leek.farmer.saveAi(aid, ais[ai]) if ai == pool + '-AI': leek.setAi(aid) except ValueError as err: print(format(err)) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def tournament(params, options): try: for leek in Pool.get(Settings(options), Pool.parse(options)): try: leek.tournament() leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def equipWeapon(params, options): try: template = params[0] for leek in Pool.get(Settings(options), Pool.parse(options)): try: wid = None for weapon in leek.farmer.weapons: if (weapon['template'] == template): wid = weapon['id'] if wid is None: leek.farmer.raiseError( 'Have any {0} available'.format(template)) leek.equipWeapon(wid) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def unequipWeapon(params, options): try: template = params[0] for leek in Pool.get(Settings(options), Pool.parse(options)): try: wid = None for weapon in leek.weapons: if (weapon['template'] == template): wid = weapon['id'] if wid is None: leek.farmer.raiseError( 'Have any {0} available'.format(template)) leek.unequipWeapon(wid) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def equipChip(params, options): try: template = params[0] for leek in Pool.get(Settings(options), Pool.parse(options)): try: wid = None for chip in leek.farmer.chips: if (chip['template'] == template): wid = chip['id'] if wid is None: leek.farmer.raiseError( 'Have any {0} available'.format(template)) leek.equipChip(wid) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def unequipChip(params, options): try: template = params[0] for leek in Pool.get(Settings(options), Pool.parse(options)): try: wid = None for chip in leek.chips: if (chip['template'] == template): wid = chip['id'] if wid is None: leek.farmer.raiseError( 'Have any {0} available'.format(template)) leek.unequipChip(wid) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def buy(params, options): item = params[0] try: for leek in Pool.get(Settings(options), Pool.parse(options)): try: for x in (leek.farmer.chips + leek.farmer.weapons): if x['template'] == item: leek.farmer.raiseError('Allready have one') leek.farmer.buy(item) leek.farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def sell(params, options): item = params[0] try: for leek in Pool.get(Settings(options), Pool.parse(options)): try: leek.farmer.sell(item) leek.farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def characteristics(params, options): bonuses = { 'life': 0, 'strength': 0, 'wisdom': 0, 'agility': 0, 'resistance': 0, 'frequency': 0, 'science': 0, 'magic': 0, 'tp': 0, 'mp': 0 } bonuses[params[1]] = params[0] try: for leek in Pool.get(Settings(options), Pool.parse(options)): try: leek.characteristics(bonuses) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) except ValueError as err: print(format(err)) def teamJoin(params, options): team = params[0] try: owner = Farmers.farmer(Settings(options), Team.getOwner(team)) owner.openTeam('true') for leek in Pool.get(Settings(options), Pool.parse(options)): try: leek.farmer.joinTeam(team) for candidacy in owner.getTeam(team)['candidacies']: if candidacy['farmer']['id'] == leek.farmer.id: owner.acceptTeamCandidacy(candidacy['id']) leek.raiseError('OK') #Ugly leek.raiseError('Can\'t find candidacy') #Ugly except ValueError as err: print(format(err)) owner.openTeam('false') except ValueError as err: print(format(err)) def teamComposition(params, options): try: leeks = Pool.get(Settings(options), Pool.parse(options)) team = leeks[0].farmer.data['team']['id'] owner = Farmers.farmer(Settings(options), Team.getOwner(team)) composition = owner.createTeamComposition(params[0])['id'] for leek in leeks: try: owner.setTeamComposition(composition, leek.id) leek.raiseError('OK') #Ugly except ValueError as err: print(format(err)) for compo in owner.getTeam(team)['compositions']: if len(compo['leeks']) == 0: owner.removeTeamComposition(compo['id']) except ValueError as err: print(format(err)) def teamTournament(params, options): try: leek = Pool.get(Settings(options), Pool.parse(options))[0] team = leek.farmer.data['team']['id'] owner = Farmers.farmer(Settings(options), Team.getOwner(team)) for composition in owner.getTeam(team)['compositions']: cleeks = [x['id'] for x in composition['leeks']] if leek.id in cleeks: owner.tournamentTeamComposition(composition['id']) leek.raiseError('OK') #Ugly leek.raiseError('Can\'t find composition') except ValueError as err: print(format(err)) def teamFight(params, options): try: leek = Pool.get(Settings(options), Pool.parse(options))[0] team = leek.farmer.data['team']['id'] for composition in leek.farmer.getTeam(team)['compositions']: cleeks = [x['id'] for x in composition['leeks']] if leek.id in cleeks: cid = composition['id'] for _ in range(params[0] if type(params[0]) is int else 20): opponents = leek.getCompositionOpponents(cid) if len(opponents) < 1: leek.raiseError('Probably, no more team fights') opid = None optalent = 20000000 for x in opponents: if x['talent'] < optalent: opid = x['id'] optalent = x['talent'] print('https://leekwars.com/fight/{0}'.format( leek.teamFight(cid, opid))) time.sleep(options['sleep']) leek.raiseError('OK') #Ugly leek.raiseError('Can\'t find composition') except ValueError as err: print(format(err)) def auto(params, options): Pool.fight([None, None], options) Pool.fight([None, 'force'], options) time.sleep(options['sleep']*10) Pool.teamFight([None], options) Pool.tournament([], options) Pool.teamTournament([], options) class Team: def create(params, options): name = params[0] try: farmer = Farmers.farmer(params[1]) farmer.createTeam(name) farmer.raiseError('OK') #Ugly except ValueError as err: print(format(err)) def get(team): req = lwapi.team.get(team) if not req.get('success', False): raise ValueError('Team {0} : {1}'.format(team, req.get('error', 'Fail'))) return req['team'] def getOwner(team): for member in Team.get(team)['members']: if member['grade'] == 'owner': return str(member['id']) raise ValueError('Team {0} : Can\'t find owner'.format(team)) class Farmer: def __init__(self, data): self.data = data['farmer'] self.token = data['token'] self.id = self.data['id'] self.name = self.data['name'] self.weapons = self.data['weapons'] self.chips = self.data['chips'] self.leeks = self.data['leeks'] self.fights = self.data['fights'] def raiseError(self, error): raise ValueError('Farmer {0} : {1}'.format(self.name, error)) def checkRequest(self, req): if not req.get('success', False): self.raiseError(req.get('error', 'Fail')) return req def buy(self, item): self.checkRequest(lwapi.market.buy_habs(item, self.token)) def sell(self, item): self.checkRequest(lwapi.market.sell_habs(item, self.token)) def getLeek(self, leek): return Leek( self.checkRequest(lwapi.leek.get_private(leek, self.token)), self) def getOpponents(self): return self.checkRequest( lwapi.garden.get_farmer_opponents(self.token))['opponents'] def getAis(self): return self.checkRequest(lwapi.ai.get_farmer_ais(self.token))['ais'] def newAi(self, folder, name): ai = self.checkRequest(lwapi.ai.new(folder, 'false', self.token))['ai'] self.checkRequest(lwapi.ai.rename(ai['id'], name, self.token)) return ai def saveAi(self, ai, script): ai = self.checkRequest(lwapi.ai.save(ai, script, self.token))['result'] if len(ai[0]) > 3: self.raiseError('Script error ('+', '.join(str(x) for x in ai[0][3:])+')') def joinTeam(self, team): self.checkRequest(lwapi.team.send_candidacy(team, self.token)) def openTeam(self, open): self.checkRequest(lwapi.team.set_opened(open, self.token)) def createTeam(self, name): self.checkRequest(lwapi.team.create(name, self.token)) self.openTeam('false') def getTeam(self, team): return self.checkRequest(lwapi.team.get_private(team, self.token))['team'] def acceptTeamCandidacy(self, candidacy): self.checkRequest(lwapi.team.accept_candidacy(candidacy, self.token)) def createTeamComposition(self, composition): return self.checkRequest( lwapi.team.create_composition(composition, self.token)) def setTeamComposition(self, composition, leek): self.checkRequest(lwapi.team.move_leek(leek, composition, self.token)) def removeTeamComposition(self, composition): self.checkRequest( lwapi.team.delete_composition(composition, self.token)) def tournamentTeamComposition(self, composition): self.checkRequest(lwapi.team.register_tournament(composition, self.token)) def getFirstLeekId(self): #NOTE: Deprecated return next(iter(self.leeks)) class Leek: def __init__(self, data, farmer): self.data = data['leek'] self.farmer = farmer self.id = self.data['id'] self.name = self.data['name'] self.weapons = self.data['weapons'] self.chips = self.data['chips'] def raiseError(self, error): raise ValueError('Leek {0} : {1}'.format(self.name, error)) def checkRequest(self, req): if not req.get('success', False): self.raiseError(req.get('error', 'Fail')) return req def getOpponents(self): return self.checkRequest( lwapi.garden.get_leek_opponents(self.id, self.farmer.token))['opponents'] def fight(self, target): return self.checkRequest( lwapi.garden.start_solo_fight(self.id, target, self.farmer.token))['fight'] def setAi(self, ai): self.checkRequest(lwapi.leek.set_ai(self.id, ai, self.farmer.token)) def tournament(self): self.checkRequest( lwapi.leek.register_tournament(self.id, self.farmer.token)) def equipWeapon(self, wid): self.checkRequest( lwapi.leek.add_weapon(self.id, wid, self.farmer.token)) def unequipWeapon(self, wid): self.checkRequest(lwapi.leek.remove_weapon(wid, self.farmer.token)) def equipChip(self, wid): self.checkRequest(lwapi.leek.add_chip(self.id, wid, self.farmer.token)) def unequipChip(self, wid): self.checkRequest(lwapi.leek.remove_chip(wid, self.farmer.token)) def characteristics(self, bonuses): self.checkRequest( lwapi.leek.spend_capital(self.id, json.dumps(bonuses), self.farmer.token)) def getCompositionOpponents(self, composition): return self.checkRequest( lwapi.garden.get_composition_opponents( composition, self.farmer.token))['opponents'] def teamFight(self, composition, target): return self.checkRequest( lwapi.garden.start_team_fight(composition, target, self.farmer.token))['fight'] # Main Program if __name__ == "__main__": #TODO find best opponant CommandTree()\ .addOption('pool', ['p', '-pool'], {'name': 'pool', 'optional': True, 'default': 'main'})\ .addOption('sleep', ['s', '-sleep'], {'name': 'sleep time', 'optional': True, 'type': int, 'min': 0, 'default': 1})\ .addOption('path', ['-path'], {'name': 'config path', 'optional': True, 'default': '*/LeekBots.json'})\ .addCommand('farmers list', 'list all farmers', Farmers.list, [{'name': 'mode', 'optional': True, 'list': [None, 'infos', 'ais', 'stuff', 'leeks']}])\ .addCommand('farmers stats', 'stats of all farmers', Farmers.stats, [{'name': 'mode', 'list': ['infos']}])\ .addCommand('farmer register', 'add a new farmer',Farmers.register, [{'name': 'login'},{'name': 'password'}])\ .addCommand('farmers buy', 'buy an item FOREACH farmers',Farmers.buy, [{'name': 'item', 'type': int}])\ .addCommand('farmers sell', 'sell an item FOREACH farmers',Farmers.sell, [{'name': 'item', 'type': int}])\ .addCommand('pools list', 'list all pools',Pools.list, [])\ .addCommand('pool create', 'create a new pool',Pool.create, [])\ .addCommand('pool register', 'add a leek to a pool',Pool.register, [{'name': 'leek'}])\ .addCommand('pool list', 'list leeks',Pool.list, [{'name': 'mode', 'optional': True, 'list': [None, 'infos', 'stuff', 'characteristics']}])\ .addCommand('pool stats', 'stats of pool\'s leeks', Pool.stats, [{'name': 'mode', 'list': ['infos', 'characteristics', 'farmers']}])\ .addCommand('pool fight', 'run solo fights', Pool.fight, [{'name': 'count', 'optional': True, 'type': int, 'min': 1, 'max': 100}, {'name': 'force', 'optional': True, 'list': [None, 'force']}])\ .addCommand('pool ais', 'import ai//.leek files and load ai//AI.leek', Pool.setupAI, [])\ .addCommand('pool tournament', 'register for solo tournament', Pool.tournament, [])\ .addCommand('pool equip weapon', 'equip a weapon', Pool.equipWeapon, [{'name': 'item', 'type': int}])\ .addCommand('pool unequip weapon', 'unequip a weapon', Pool.unequipWeapon, [{'name': 'item', 'type': int}])\ .addCommand('pool equip chip', 'equip a chip', Pool.equipChip, [{'name': 'item', 'type': int}])\ .addCommand('pool unequip chip', 'unequip a chip', Pool.unequipChip, [{'name': 'item', 'type': int}])\ .addCommand('pool characteristics', 'buy characteristics', Pool.characteristics, [{'name': 'count', 'type': int}, {'name': 'type', 'list': ['life', 'strength', 'wisdom', 'agility', 'resistance', 'science', 'magic', 'tp', 'mp', 'frequency']}])\ .addCommand('pool buy', 'buy an item',Pool.buy, [{'name': 'item', 'type': int}])\ .addCommand('pool sell', 'sell an item',Pool.sell, [{'name': 'item', 'type': int}])\ .addCommand('pool team join', 'join a team', Pool.teamJoin, [{'name': 'team', 'type': int}])\ .addCommand('pool team composition', 'create a composition. must be ownered by a register farmer', Pool.teamComposition, [{'name': 'composition'}])\ .addCommand('pool team tournament', 'register composition for team tournament. based on first farmer team', Pool.teamTournament, [])\ .addCommand('pool team fight', 'run team fights', Pool.teamFight, [{'name': 'count', 'optional': True, 'type': int, 'min': 1, 'max': 20}])\ .addCommand('pool auto', 'run "fight, fight force, team fight, tournament, team tournament"', Pool.auto, [])\ .addCommand('team create', 'create a team', Team.create, [{'name': 'name'}, {'name': 'owner'}])\ .parse(sys.argv) #.addOption('farmer', ['f', '-farmer'], {'name': 'farmer', 'optional': True})\