Support None parameters

Author: Kareem Zidane

setup.py

@@ -16,5 +16,5 @@
     package_dir={"": "src"},
     packages=["cs50"],
     url="https://github.com/cs50/python-cs50",
-    version="7.0.0"
+    version="7.0.1"
 )

src/cs50/_sql_sanitizer.py

@@ -66,9 +66,7 @@ def escape(self, value):
             return sqlparse.sql.Token(sqlparse.tokens.String, string_processor(value))
 
         if value is None:
-            return sqlparse.sql.Token(
-                sqlparse.tokens.Keyword,
-                sqlalchemy.types.NullType().literal_processor(self._dialect)(value))
+            return sqlparse.sql.Token(sqlparse.tokens.Keyword, "NULL")
 
         raise RuntimeError(f"unsupported value: {value}")
 

tests/sql.py

@@ -132,19 +132,10 @@ def test_identifier_case(self):
         self.assertIn("count", self.db.execute("SELECT 1 AS count")[0])
 
     def tearDown(self):
-        self.db.execute("DROP TABLE cs50")
+        self.db.execute("DROP TABLE IF EXISTS cs50")
         self.db.execute("DROP TABLE IF EXISTS foo")
         self.db.execute("DROP TABLE IF EXISTS bar")
 
-    @classmethod
-    def tearDownClass(self):
-        try:
-            self.db.execute("DROP TABLE IF EXISTS cs50")
-        except Warning as e:
-            # suppress "unknown table"
-            if not str(e).startswith("(1051"):
-                raise e
-
 class MySQLTests(SQLTests):
     @classmethod
     def setUpClass(self):
@@ -317,6 +308,10 @@ def test_numeric(self):
     def test_cte(self):
         self.assertEqual(self.db.execute("WITH foo AS ( SELECT 1 AS bar ) SELECT bar FROM foo"), [{"bar": 1}])
 
+    def test_none(self):
+        self.db.execute("CREATE TABLE foo (val INTEGER)")
+        self.db.execute("SELECT * FROM foo WHERE val = ?", None)
+
 if __name__ == "__main__":
     suite = unittest.TestSuite([
         unittest.TestLoader().loadTestsFromTestCase(SQLiteTests),