pgserviceparser 2.2.0__tar.gz → 2.3.0__tar.gz

{pgserviceparser-2.2.0 → pgserviceparser-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: pgserviceparser
-Version: 2.2.0
+Version: 2.3.0
 Summary: A package parsing the PostgreSQL connection service file
 Author-email: David Signer <info@opengis.ch>, Julien Moura <julien.moura@oslandia.com>, Germán Carrillo <info@opengis.ch>, Denis Rouzaud <info@opengis.ch>
 License: MIT License

{pgserviceparser-2.2.0 → pgserviceparser-2.3.0}/docs/mkdocs.yml

@@ -12,4 +12,4 @@ markdown_extensions:
 
 nav:
   - pgserviceparser: index.md
-  - exceptions: excpetions.md
+  - exceptions: exceptions.md

{pgserviceparser-2.2.0 → pgserviceparser-2.3.0}/pgserviceparser/__init__.py

@@ -6,12 +6,15 @@ from pathlib import Path
 from typing import Optional
 
 # package
-from pgserviceparser.exceptions import ServiceFileNotFound, ServiceNotFound
+from .exceptions import ServiceFileNotFound, ServiceNotFound
 
 
-def conf_path() -> Path:
+def conf_path(create_if_missing: Optional[bool] = False) -> Path:
     """Returns the path found for the pg_service.conf on the system as string.
 
+    Args:
+        create_if_missing: Whether to create the file (and eventually its parent folders) if the file does not exist.
+
     Returns:
         path to the pg_service.conf file as string
     """
@@ -26,6 +29,14 @@ def conf_path() -> Path:
         else:  # Linux or Darwin (Mac)
             pg_config_path = Path("~/.pg_service.conf").expanduser()
 
+    if create_if_missing and not pg_config_path.exists():
+        # Make sure all parent directories exist
+        if not pg_config_path.parent.exists():
+            pg_config_path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Finally, make sure the file itself exists
+        Path.touch(pg_config_path)
+
     return pg_config_path
 
 

{pgserviceparser-2.2.0 → pgserviceparser-2.3.0}/pgserviceparser.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: pgserviceparser
-Version: 2.2.0
+Version: 2.3.0
 Summary: A package parsing the PostgreSQL connection service file
 Author-email: David Signer <info@opengis.ch>, Julien Moura <julien.moura@oslandia.com>, Germán Carrillo <info@opengis.ch>, Denis Rouzaud <info@opengis.ch>
 License: MIT License

{pgserviceparser-2.2.0 → pgserviceparser-2.3.0}/test/test_lib.py

@@ -123,6 +123,25 @@ class TestLib(unittest.TestCase):
         self.assertIn("service_tmp", service_names())
         remove_service("service_tmp")
 
+    def test_missing_file(self):
+        another_service_file_path = PGSERVICEPARSER_SRC_PATH / "test" / "data" / "new_folder" / "pgservice.conf"
+        os.environ["PGSERVICEFILE"] = str(another_service_file_path)
+
+        self.assertEqual(another_service_file_path, conf_path())
+        self.assertFalse(another_service_file_path.exists())
+        self.assertFalse(another_service_file_path.parent.exists())
+
+        self.assertEqual(another_service_file_path, conf_path(create_if_missing=True))
+        self.assertTrue(another_service_file_path.exists())
+        self.assertTrue(another_service_file_path.parent.exists())
+
+        os.environ["PGSERVICEFILE"] = str(self.service_file_path)
+
+    def tearDown(self):
+        new_folder_path = self.service_file_path.parent / "new_folder"
+        if new_folder_path.exists():
+            shutil.rmtree(new_folder_path)
+
 
 if __name__ == "__main__":
     unittest.main()