me
/
LeekBots
Archived
1
0
Fork 0
Code Releases Activity
Browse Source

Add buy option and formating

master
sheychen 4 years ago
parent
04519ae217
commit
c4abd79970
2 changed files with 100 additions and 77 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 49
      CommandTree.py
  2. 128
      LeekBots.py

49
CommandTree.py
Unescape View File

@ -27,9 +27,9 @@ class CommandTree:
name += ':'
if data is None:
if not param.get('optional', False):
print('Wrong params count in "{0}", {1} isn\'t optional.'.
format(path, name))
return False
raise ValueError(
'Wrong params count in "{0}", {1} isn\'t optional.'.
format(path, name))
else:
pdef = param.get('default')
if not pdef is None:
@ -40,41 +40,39 @@ class CommandTree:
try:
data = ptype(data)
except (TypeError, ValueError):
print('Wrong type in "{0}", {1}"{2}" must be an {3}.'.
format(path, name, data, ptype.__name__))
return False
raise ValueError(
'Wrong type in "{0}", {1}"{2}" must be an {3}.'.
format(path, name, data, ptype.__name__))
if ptype == int:
pmin = param.get('min')
if type(pmin) is int:
if data < pmin:
print(
raise ValueError(
'Wrong value in "{0}", {1}"{2}" must be >= {3}.'.
format(path, name, data, pmin))
return False
pmax = param.get('max')
if type(pmax) is int:
if data > pmax:
print(
raise ValueError(
'Wrong value in "{0}", {1}"{2}" must be <= {3}.'.
format(path, name, data, pmax))
return False
plist = param.get('list')
if type(plist) is list:
if not data in plist:
print(
raise ValueError(
'Wrong value in "{0}", {1}"{2}" must be one of ({3}).'.
format(path, name, data, ', '.join(
str(x) for x in plist)))
return False
else:
print('"{0}" doesn\'t accepts so many params.'.format(path))
return False
raise ValueError(
'"{0}" doesn\'t accepts so many params.'.format(path))
return data
def runCommand(self, path, args, options):
command = self.commands[path]
params = []
for data, param in itertools.zip_longest(args[len(path.split()):], command['params']):
for data, param in itertools.zip_longest(args[len(path.split()):],
command['params']):
data = CommandTree.checkOption(data, param, path)
if data is False:
return
@ -109,9 +107,15 @@ class CommandTree:
option = path.pop(0)[1:]
for key in self.options:
if option in self.options[key]['names']:
data = CommandTree.checkOption(
path.pop(0), self.options[key]['check'], option)
if data is False:
try:
value = True
if not self.options[key]['check'].get('type') == bool:
value = path.pop(0)
data = CommandTree.checkOption(
value, self.options[key]['check'],
option)
except ValueError as err:
print(format(err))
return
options[key] = data
option = None
@ -119,11 +123,12 @@ class CommandTree:
print('Unknown option "{0}"'.format(option))
return
for key in self.options:
data = CommandTree.checkOption(
options.get(key), self.options[key]['check'], key)
if data is False:
try:
options[key] = CommandTree.checkOption(
options.get(key), self.options[key]['check'], key)
except ValueError as err:
print(format(err))
return
options[key] = data
commands = list(self.commands.keys())
if len(path) > 0:
run = self.runCommand

128
LeekBots.py
Unescape View File

@ -211,9 +211,10 @@ class Farmers:
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])))
print('{0} : {1}, {2}, {3}'.format(
field, min(fields[field]),
int(sum(fields[field]) / len(fields[field])),
max(fields[field])))
@staticmethod
def register(params, options):
@ -264,8 +265,7 @@ class Farmers:
try:
random.seed()
farmer = Farmers.farmer(Settings(options), Farmers.parse(options))
for _ in range(params[0]
if type(params[0]) is int else 20):
for _ in range(params[0] if type(params[0]) is int else 20):
opponents = farmer.getOpponents()
if len(opponents) < 1:
farmer.raiseError('Probably, no more farmer fights')
@ -279,13 +279,10 @@ class Farmers:
else:
optalent = 0
for x in opponents:
if (
options['selection-mode'] == 'worst'
and optalent > x['talent']
) or (
not options['selection-mode'] == 'worst'
and optalent < x['talent']
):
if (options['selection-mode'] == 'worst'
and optalent > x['talent']) or (
not options['selection-mode'] == 'worst'
and optalent < x['talent']):
opid = x['id']
optalent = x['talent']
@ -319,16 +316,17 @@ class Farmers:
itemsCount = 0
haveTooMuch = True
while haveTooMuch:
print('With '+str(itemsCount)+' items')
print('With ' + str(itemsCount) + ' items')
haveTooMuch = False
for combination in itertools.combinations_with_replacement(buys, itemsCount):
for combination in itertools.combinations_with_replacement(
buys, itemsCount):
currentHabs = sum(buys[op] for op in combination)
if currentHabs == targetHabs:
print(combination)
return
elif currentHabs > targetHabs:
haveTooMuch = True
itemsCount+=1
itemsCount += 1
farmer.raiseError('OK') #Ugly
except ValueError as err:
@ -484,9 +482,10 @@ class Pool:
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])))
print('{0} : {1}, {2}, {3}'.format(
field, min(fields[field]),
int(sum(fields[field]) / len(fields[field])),
max(fields[field])))
@staticmethod
def items(params, options):
@ -495,7 +494,7 @@ class Pool:
if mode == None:
mode = 'all'
try:
i=0
i = 0
leeks = Pool.get(Settings(options), Pool.parse(options))
for leek in leeks:
itemList = []
@ -507,16 +506,16 @@ class Pool:
itemList += leek.farmer.potions
for item in itemList:
if not item['template'] in items:
items[item['template']] = [0]*len(leeks)
items[item['template']] = [0] * len(leeks)
items[item['template']][i] += 1
i+=1
i += 1
except ValueError as err:
print(format(err))
print('item : min, avg, max')
for field in items:
print('{0} : {1}, {2}, {3}'.format(Items.getName(field), min(
items[field]), int(sum(items[field]) / len(items[field])),
max(items[field])))
print('{0} : {1}, {2}, {3}'.format(
Items.getName(field), min(items[field]),
int(sum(items[field]) / len(items[field])), max(items[field])))
@staticmethod
def fight(params, options):
@ -528,7 +527,8 @@ class Pool:
force = (params[1] == 'force')
for leek in leeks:
try:
fights = params[0] if type(params[0]) is int else leek.farmer.fights
fights = params[0] if type(
params[0]) is int else leek.farmer.fights
if fights < 1:
leek.raiseError('No more fights')
@ -556,13 +556,10 @@ class Pool:
for x in opponents:
if force or not x['id'] in leeksids:
score = x['level'] * x['talent']
if (
options['selection-mode'] == 'worst'
and score < opscore
) or (
not options['selection-mode'] == 'worst'
and score > opscore
):
if (options['selection-mode'] == 'worst'
and score < opscore
) or (not options['selection-mode'] ==
'worst' and score > opscore):
opid = x['id']
opscore = score
if opid is None:
@ -645,8 +642,13 @@ class Pool:
wid = weapon['id']
if wid is None:
leek.farmer.raiseError(
'Have any {0} available'.format(template))
if options['buy']:
print(leek.farmer.name + ' buy')
wid = leek.farmer.buy(template)
else:
leek.farmer.raiseError(
'Have any {0} available. (use --buy)'.format(
template))
leek.equipWeapon(wid)
leek.raiseError('OK') #Ugly
@ -689,8 +691,13 @@ class Pool:
wid = chip['id']
if wid is None:
leek.farmer.raiseError(
'Have any {0} available'.format(template))
if options['buy']:
print(leek.farmer.name + ' buy')
wid = leek.farmer.buy(template)
else:
leek.farmer.raiseError(
'Have any {0} available. (use --buy)'.format(
template))
leek.equipChip(wid)
leek.raiseError('OK') #Ugly
@ -733,8 +740,13 @@ class Pool:
pid = potion['id']
if pid is None:
leek.farmer.raiseError(
'Have any {0} available'.format(template))
if options['buy']:
print(leek.farmer.name + ' buy')
pid = leek.farmer.buy(template)
else:
leek.farmer.raiseError(
'Have any {0} available. (use --buy)'.format(
template))
leek.usePotion(pid)
leek.raiseError('OK') #Ugly
@ -802,7 +814,7 @@ class Pool:
def teamJoin(params, options):
team = params[0]
try:
owner = Farmers.farmer(Settings(options), Team.getOwner(team))
owner = Farmers.farmer(Settings(options), Team.getOwner(team))
owner.openTeam('true')
for leek in Pool.get(Settings(options), Pool.parse(options)):
try:
@ -847,7 +859,7 @@ class Pool:
cleeks = [x['id'] for x in composition['leeks']]
if leek.id in cleeks:
owner.tournamentTeamComposition(composition['id'])
leek.raiseError('OK') #Ugly
leek.raiseError('OK') #Ugly
leek.raiseError('Can\'t find composition')
except ValueError as err:
print(format(err))
@ -873,7 +885,7 @@ class Pool:
if leek.id in cleeks:
cid = composition['id']
for _ in range(params[0]
if type(params[0]) is int else 20):
if type(params[0]) is int else 20):
opponents = leek.getCompositionOpponents(cid)
if len(opponents) < 1:
leek.raiseError('Probably, no more team fights')
@ -888,12 +900,10 @@ class Pool:
optalent = 0
for x in opponents:
if (
options['selection-mode'] == 'worst'
and optalent > x['talent']
) or (
not options['selection-mode'] == 'worst'
and optalent < x['talent']
):
options['selection-mode'] == 'worst'
and optalent > x['talent']
) or (not options['selection-mode'] == 'worst'
and optalent < x['talent']):
opid = x['id']
optalent = x['talent']
@ -909,7 +919,7 @@ class Pool:
def auto(params, options):
Pool.fight([None, None], options)
Pool.fight([None, 'force'], options)
time.sleep(options['sleep']*10)
time.sleep(options['sleep'] * 10)
Pool.teamFight([None], options)
Pool.tournament([], options)
Pool.teamTournament([], options)
@ -998,13 +1008,15 @@ class Farmer:
return req
def buy(self, item):
self.checkRequest(lwapi.market.buy_habs(item, self.token))
return self.checkRequest(lwapi.market.buy_habs(item,
self.token))['item']
def sell(self, item):
self.checkRequest(lwapi.market.sell_habs(item, self.token))
def market(self):
return self.checkRequest(lwapi.market.get_item_templates(self.token))['items']
return self.checkRequest(lwapi.market.get_item_templates(
self.token))['items']
def getLeek(self, leek):
return Leek(
@ -1025,7 +1037,8 @@ class Farmer:
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:])+')')
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))
@ -1056,19 +1069,22 @@ class Farmer:
lwapi.team.delete_composition(composition, self.token))
def tournamentTeamComposition(self, composition):
self.checkRequest(lwapi.team.register_tournament(composition, self.token))
self.checkRequest(
lwapi.team.register_tournament(composition, self.token))
def tournament(self):
self.checkRequest(lwapi.farmer.register_tournament(self.token))
def fight(self, target):
return self.checkRequest(lwapi.garden.start_farmer_fight(target, self.token))
return self.checkRequest(
lwapi.garden.start_farmer_fight(target, self.token))
def avatar(self, avatar):
return self.checkRequest(lwapi.farmer.set_avatar(avatar, self.token))
def teamEmblem(self, team, emblem):
return self.checkRequest(lwapi.team.set_emblem(team, emblem, self.token))
return self.checkRequest(
lwapi.team.set_emblem(team, emblem, self.token))
def createLeek(self, name):
return self.checkRequest(lwapi.leek.create(name, self.token))
@ -1127,7 +1143,8 @@ class Leek:
self.checkRequest(lwapi.leek.remove_chip(wid, self.farmer.token))
def usePotion(self, pid):
self.checkRequest(lwapi.leek.use_potion(self.id, pid, self.farmer.token))
self.checkRequest(
lwapi.leek.use_potion(self.id, pid, self.farmer.token))
def characteristics(self, bonuses):
self.checkRequest(
@ -1152,6 +1169,7 @@ if __name__ == "__main__":
.addOption('farmer', ['f', '-farmer'], {'name': 'farmer (id)', 'optional': True})\
.addOption('selection-mode', ['sm', '-selection-mode'], {'name': 'selection mode', 'optional': True, 'list': ['random', 'best', 'worst'], 'default': 'worst'})\
.addOption('sleep', ['s', '-sleep'], {'name': 'sleep time', 'optional': True, 'type': int, 'min': 0, 'default': 1})\
.addOption('buy', ['b', '-buy'], {'name': 'should buy', 'optional': True, 'type': bool, 'default': False})\
.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']}])\