diff options
Diffstat (limited to 'modules/ships.py')
| -rw-r--r-- | modules/ships.py | 8 |
1 files changed, 4 insertions, 4 deletions
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() |