diff --git a/calefaction/auth.py b/calefaction/auth.py index c67259b..1521229 100644 --- a/calefaction/auth.py +++ b/calefaction/auth.py @@ -207,7 +207,7 @@ class AuthManager: return False self._debug("Setting style to %s for char id=%d", style, cid) - g.db.set_character_style(cid, style) + g.db.update_character(cid, "style", style) if hasattr(g, "_character_props"): delattr(g, "_character_props") diff --git a/calefaction/database.py b/calefaction/database.py index 073ef50..6ba7349 100644 --- a/calefaction/database.py +++ b/calefaction/database.py @@ -148,11 +148,13 @@ class Database: res = self._conn.execute(query, (cid,)).fetchall() return {"name": res[0][0], "style": res[0][1]} if res else {} - def set_character_style(self, cid, style): - """Update a character's style setting.""" + def update_character(self, cid, prop, value): + """Update a property for the given character.""" + props = {"name": "character_name", "style": "character_style"} + field = props[prop] with self._conn as conn: - conn.execute("""UPDATE character SET character_style = ? - WHERE character_id = ?""", (style, cid)) + conn.execute("""UPDATE character SET {} = ? + WHERE character_id = ?""".format(field), (value, cid)) def set_auth(self, cid, token, expires, refresh): """Set the authentication info for the given character."""