plomlompom.com Git - plomrogue2-experiments/blob - new2/plomrogue/io.py

   1 import queue
   2
   3
   4
   5 class GameIO():
   6
   7     def __init__(self, game, save_file='savefile'):
   8         from plomrogue.parser import Parser
   9         self.clients = {}
  10         self.parser = Parser(game)
  11         self.game = game
  12         self.save_file = save_file
  13
  14     def loop(self, q):
  15         """Handle commands coming through queue q, run game, send results back."""
  16         while True:
  17             try:
  18                 command, connection_id = q.get(timeout=0.001)
  19                 self.handle_input(connection_id, command)
  20             except queue.Empty:
  21                 self.game.run_tick()
  22
  23     def run_loop_with_server(self, port, server_class):
  24         """Run connection of server talking to clients and game IO loop.
  25
  26         We have a server of self.server_class and we have the game IO loop,
  27         a thread running self.loop. Both communicate with each other via a
  28         queue.Queue. While the server may spawn parallel threads to many
  29         clients, the IO loop works sequentially through game commands
  30         received from the server's client connections.  A processed command
  31         may trigger messages to the commanding client or to all clients,
  32         delivered from the IO loop to the server via the queue.
  33
  34         """
  35         import threading
  36         q = queue.Queue()
  37         c = threading.Thread(target=self.loop, daemon=True, args=(q,))
  38         c.start()
  39         self.server = server_class(q, port)
  40         try:
  41             self.server.serve_forever()
  42         except KeyboardInterrupt:
  43             pass
  44         finally:
  45             print('Killing server')
  46             self.server.server_close()
  47
  48     def handle_input(self, input_, connection_id=None):
  49         """Process input_ to command grammar, call command handler if found."""
  50         from inspect import signature
  51         from plomrogue.errors import GameError, ArgError
  52         from plomrogue.misc import quote
  53
  54         def answer(connection_id, msg):
  55             if connection_id:
  56                 self.send(msg, connection_id)
  57             else:
  58                 print(msg)
  59
  60         try:
  61             command, args = self.parser.parse(input_)
  62             if command is None:
  63                 answer(connection_id, 'UNHANDLED_INPUT')
  64             else:
  65                 if 'connection_id' in list(signature(command).parameters):
  66                     command(*args, connection_id=connection_id)
  67                 else:
  68                     command(*args)
  69                     #if store and not hasattr(command, 'dont_save'):
  70                     #    with open(self.game_file_name, 'a') as f:
  71                     #        f.write(input_ + '\n')
  72         except ArgError as e:
  73             answer(connection_id, 'ARGUMENT_ERROR ' + quote(str(e)))
  74         except GameError as e:
  75             answer(connection_id, 'GAME_ERROR ' + quote(str(e)))
  76
  77     def send(self, msg, connection_id=None):
  78         """Send message msg to server's client(s).
  79
  80         If a specific client is identified by connection_id, only
  81         sends msg to that one. Else, sends it to all clients.
  82
  83         """
  84         if connection_id:
  85             self.server.clients[connection_id].put(msg)
  86         else:
  87             for c in self.server.clients.values():
  88                 c.put(msg)