Add the PSL constructor for loading list from file

Author: rmisev

src/upa_url_bind.cpp

@@ -204,6 +204,11 @@ NB_MODULE(upa_url, m) {
         .def("finalize", &public_suffix_list_py::finalize)
 
         // Load Public Suffix List from file
+        .def("__init__", [](public_suffix_list_py* t, std::string_view filename) {
+                new (t) public_suffix_list_py{};
+                if (!t->load(filename))
+                    throw std::runtime_error("Error loading Public Suffix List from file.");
+            }, nb::arg("filename"))
         .def_static("load", [](std::string_view filename)
             -> std::optional<public_suffix_list_py> {
                 public_suffix_list_py psl;

tests/test_psl.py

@@ -25,6 +25,17 @@ def test_psl_push(self):
         psl.push('^.com')
         self.assertFalse(psl.finalize())
 
+    def test_psl_construct(self):
+        # Use constructor to load the list
+        dir = os.path.dirname(os.path.realpath(__file__))
+        psl = PSL(os.path.join(dir, 'PSL.dat'))
+        # public suffix
+        self.assertEqual(psl.public_suffix('any.org.uk'), 'org.uk')
+
+        # Load error
+        with self.assertRaises(RuntimeError):
+            PSL(os.path.join(dir, 'no_such_file.dat'))
+
     def test_psl(self):
         # Load list
         dir = os.path.dirname(os.path.realpath(__file__))