add cleanup code for storage classes
This commit is contained in:
+9
-1
@@ -4,7 +4,7 @@ from server.SnakeBuilder import SnakeBuilder
|
||||
|
||||
from server.storage.StorageLoader import StorageLoader
|
||||
|
||||
from flask import Flask
|
||||
from flask import Flask, jsonify
|
||||
from flask import request
|
||||
import logging, json, os, re
|
||||
|
||||
@@ -54,6 +54,10 @@ class Server:
|
||||
"server", "battlesnake/github/starter-snake-python"
|
||||
)
|
||||
return response
|
||||
|
||||
@self.app.post("/cleanup")
|
||||
def cleanup():
|
||||
self._cleanup()
|
||||
|
||||
def run(self, host:str="0.0.0.0", port:str="8000", debug:bool=False):
|
||||
logging.getLogger("werkzeug").setLevel(logging.ERROR)
|
||||
@@ -152,3 +156,7 @@ class Server:
|
||||
|
||||
print("GAME ENDED: Winner is", [ x["name"] for x in game_state["board"]['snakes']])
|
||||
self._delete_game_board(game_state)
|
||||
|
||||
def _cleanup(self):
|
||||
storage = StorageLoader.build(self.storage_type)
|
||||
return jsonify(storage.cleanup())
|
||||
|
||||
Reference in New Issue
Block a user