From 26086596b13d977c9558cbdb07a166ec414d5dbb Mon Sep 17 00:00:00 2001
From: "WQY\\qiong" <qiongyu.wang@outlook.com>
Date: Sun, 27 Nov 2022 00:10:49 +0800
Subject: [PATCH] Fix demands undo bug

---
 api/s9_demands.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/s9_demands.py b/api/s9_demands.py
index 83593f7..8a01848 100644
--- a/api/s9_demands.py
+++ b/api/s9_demands.py
@@ -44,8 +44,8 @@ def set_demand_cache(name: str, cs: ChangeSet) -> BatchSqlChangeSet:
     _undo_sql = f"delete from demands where junction = {f_junction};"
     for r in old['demands']:
         demand = float(r['demand'])
-        pattern = str(r['pattern'])
-        category = str(r['category'])
+        pattern = str(r['pattern']) if 'pattern' in r and r['pattern'] != None else None
+        category = str(r['category']) if 'category' in r and r['category'] != None else None
         f_demand = demand
         f_pattern = f"'{pattern}'" if pattern != None else 'null'
         f_category = f"'{category}'" if category != None else 'null'