|
|
@@ -59,14 +59,17 @@ class Player:
|
|
|
space += spec['levels'][self.buildings[building] - 1]['capacity']
|
|
|
return space
|
|
|
|
|
|
+ def add_resource(self, resource, amount):
|
|
|
+ self.resources[resource] += amount
|
|
|
+ self.resources[resource] = min(self.get_storage_space(resource), self.resources[resource])
|
|
|
+
|
|
|
def update(self, tick_length):
|
|
|
self.jobs = [job for job in self.jobs if not job.check_finish()]
|
|
|
# Resource generation
|
|
|
for building in self.buildings.keys():
|
|
|
spec = core.config['building'][building]
|
|
|
if 'production' in spec and self.buildings[building] > 0:
|
|
|
- self.resources[spec['production']] += spec['levels'][self.buildings[building] - 1]['rate'] * tick_length
|
|
|
- self.resources[spec['production']] = min(self.get_storage_space(spec['production']), self.resources[spec['production']])
|
|
|
+ self.add_resource(spec['production'], spec['levels'][self.buildings[building] - 1]['rate'] * tick_length)
|
|
|
|
|
|
self.ws.send_json({
|
|
|
'username': self.username,
|
