From 569469eac4b454f1887f892e308922cd460c89dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lu=C3=ADs=20Murta?= <luis@murta.dev>
Date: Mon, 7 Jun 2021 20:33:34 +0100
Subject: [PATCH] [Fix] Category SELECT can now get Null values

---
 pfbudget/database.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pfbudget/database.py b/pfbudget/database.py
index 51109cb..ef70148 100644
--- a/pfbudget/database.py
+++ b/pfbudget/database.py
@@ -75,7 +75,7 @@ WHERE date BETWEEN (?) AND (?)
 SELECT_TRANSACTIONS_BY_CATEGORY = """
 SELECT *
 FROM transactions
-WHERE category = (?)
+WHERE category IS (?)
 """
 
 SELECT_TRANSACTION_BY_PERIOD = """
@@ -183,6 +183,10 @@ class DBManager:
         logger.info(f"Get transactions by {period}")
         return self.__execute(SELECT_TRANSACTION_BY_PERIOD, period)
 
+    def get_uncategorized_transactions(self):
+        logger.info("Get uncategorized transactions")
+        return self.get_category(None)
+
     def export(self):
         filename = pathlib.Path(
             "@".join([self.db, datetime.datetime.now().isoformat()])