From a994b822a4231061153c6b27d8c5d7307b59f8b0 Mon Sep 17 00:00:00 2001 From: Botond Hende Date: Sun, 1 Sep 2024 23:33:15 +0200 Subject: ship task setting from request body --- modules/ships.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'modules/ships.py') diff --git a/modules/ships.py b/modules/ships.py index e35cdd2..05ce6c1 100644 --- a/modules/ships.py +++ b/modules/ships.py @@ -1,6 +1,6 @@ from typing import Dict -from . import task_type +from . import task_types from .database import cursor, conn @@ -20,10 +20,10 @@ class Ship: row = cursor.fetchone() if row is None: cursor.execute("INSERT INTO ships (symbol, task, params, name) VALUES (?, ?, ?, ?)", - (self.symbol, task_type.IDLE, None, self.symbol)) + (self.symbol, task_types.IDLE, None, self.symbol)) conn.commit() - self.task = task_type.IDLE + self.task = task_types.IDLE self.task = self.symbol else: self.name = row[0] @@ -31,7 +31,7 @@ class Ship: def set_task(self, task): self.task = task - if task != task_type.ERROR: + if task != task_types.ERROR: cursor.execute("UPDATE ships SET task = ? WHERE symbol = ?", (task, self.symbol)) conn.commit() -- cgit v1.2.3-70-g09d2