pyglove 0.4.5.dev202503190809__py3-none-any.whl → 0.4.5.dev202503210809__py3-none-any.whl

pyglove/core/io/file_system.py

@@ -76,6 +76,10 @@ class FileSystem(metaclass=abc.ABCMeta):
   ) -> File:
     """Opens a file with a path."""
 
+  @abc.abstractmethod
+  def chmod(self, path: Union[str, os.PathLike[str]], mode: int) -> None:
+    """Changes the permission of a file."""
+
   @abc.abstractmethod
   def exists(self, path: Union[str, os.PathLike[str]]) -> bool:
     """Returns True if a path exists."""
@@ -167,6 +171,9 @@ class StdFileSystem(FileSystem):
   ) -> File:
     return StdFile(io.open(path, mode, **kwargs))
 
+  def chmod(self, path: Union[str, os.PathLike[str]], mode: int) -> None:
+    os.chmod(path, mode)
+
   def exists(self, path: Union[str, os.PathLike[str]]) -> bool:
     return os.path.exists(path)
 
@@ -272,6 +279,10 @@ class MemoryFileSystem(FileSystem):
       raise FileNotFoundError(path)
     return file
 
+  def chmod(self, path: Union[str, os.PathLike[str]], mode: int) -> None:
+    # No-op.
+    del path, mode
+
   def exists(self, path: Union[str, os.PathLike[str]]) -> bool:
     return self._locate(path) is not None
 
@@ -411,6 +422,11 @@ def open(path: Union[str, os.PathLike[str]], mode: str = 'r', **kwargs) -> File:
   return _fs.get(path).open(path, mode, **kwargs)
 
 
+def chmod(path: Union[str, os.PathLike[str]], mode: int) -> None:
+  """Changes the permission of a file."""
+  _fs.get(path).chmod(path, mode)
+
+
 def readfile(
     path: Union[str, os.PathLike[str]],
     mode: str = 'r',
@@ -432,11 +448,14 @@ def writefile(
     content: Union[str, bytes],
     *,
     mode: str = 'w',
+    perms: Optional[int] = 0o664,   # Default to world-readable.
     **kwargs,
 ) -> None:
   """Writes content to a file."""
   with _fs.get(path).open(path, mode=mode, **kwargs) as f:
     f.write(content)
+  if perms is not None:
+    chmod(path, perms)
 
 
 def rm(path: Union[str, os.PathLike[str]]) -> None:

pyglove/core/io/sequence.py

@@ -26,9 +26,11 @@ class Sequence(metaclass=abc.ABCMeta):
 
   def __init__(
       self,
+      perms: Optional[int] = None,
       serializer: Optional[Callable[[Any], Union[bytes, str]]] = None,
       deserializer: Optional[Callable[[Union[bytes, str]], Any]] = None
   ):
+    self._perms = perms
     self._serializer = serializer
     self._deserializer = deserializer
 
@@ -88,6 +90,7 @@ class SequenceIO(metaclass=abc.ABCMeta):
       path: Union[str, os.PathLike[str]],
       mode: str,
       *,
+      perms: Optional[int],
       serializer: Optional[Callable[[Any], Union[bytes, str]]],
       deserializer: Optional[Callable[[Union[bytes, str]], Any]],
       **kwargs
@@ -130,6 +133,7 @@ def open_sequence(
     path: Union[str, os.PathLike[str]],
     mode: str = 'r',
     *,
+    perms: Optional[int] = 0o664,  # Default to world-readable.
     serializer: Optional[
         Callable[[Any], Union[bytes, str]]
     ] = None,
@@ -143,6 +147,7 @@ def open_sequence(
   Args:
     path: The path to the sequence.
     mode: The mode of the sequence.
+    perms: (Optional) The permissions of the sequence.
     serializer: (Optional) A serializer function for converting a structured
       object to a string or bytes.
     deserializer: (Optional) A deserializer function for converting a string or
@@ -158,7 +163,7 @@ def open_sequence(
     if make_dirs_if_not_exist:
       file_system.mkdirs(parent_dir, exist_ok=True)
   return _registry.get(path).open(
-      path, mode, serializer=serializer, deserializer=deserializer
+      path, mode, perms=perms, serializer=serializer, deserializer=deserializer
   )
 
 
@@ -170,10 +175,12 @@ class MemorySequence(Sequence):
       path: str,
       mode: str,
       records: list[Union[str, bytes]],
+      *,
+      perms: Optional[int],
       serializer: Optional[Callable[[Any], Union[bytes, str]]],
       deserializer: Optional[Callable[[Union[bytes, str]], Any]]
   ):
-    super().__init__(serializer, deserializer)
+    super().__init__(perms, serializer, deserializer)
     self._path = path
     self._mode = mode
     self._records = records
@@ -226,6 +233,7 @@ class MemorySequenceIO(SequenceIO):
       path: Union[str, os.PathLike[str]],
       mode: str,
       *,
+      perms: Optional[int],
       serializer: Optional[Callable[[Any], Union[bytes, str]]],
       deserializer: Optional[Callable[[Union[bytes, str]], Any]],
       **kwargs
@@ -236,7 +244,7 @@ class MemorySequenceIO(SequenceIO):
       self._root[path] = []
     return MemorySequence(
         path, mode, self._root[path],
-        serializer=serializer, deserializer=deserializer
+        perms=perms, serializer=serializer, deserializer=deserializer
     )
 
 
@@ -248,12 +256,16 @@ class LineSequence(Sequence):
 
   def __init__(
       self,
-      file: file_system.File,
+      path: str,
+      mode: str,
+      perms: Optional[int],
       serializer: Optional[Callable[[Any], Union[bytes, str]]],
       deserializer: Optional[Callable[[Union[bytes, str]], Any]],
   ) -> None:
-    super().__init__(serializer, deserializer)
-    self._file = file
+    super().__init__(perms, serializer, deserializer)
+    self._path = path
+    self._mode = mode
+    self._file = file_system.open(path, mode)
 
   def __len__(self):
     raise NotImplementedError(
@@ -277,6 +289,8 @@ class LineSequence(Sequence):
 
   def close(self) -> None:
     self._file.close()
+    if ('w' in self._mode or 'a' in self._mode) and self._perms is not None:
+      file_system.chmod(self._path, self._perms)
 
 
 class LineSequenceIO(SequenceIO):
@@ -287,13 +301,12 @@ class LineSequenceIO(SequenceIO):
       path: Union[str, os.PathLike[str]],
       mode: str,
       *,
+      perms: Optional[int],
       serializer: Optional[Callable[[Any], Union[bytes, str]]],
       deserializer: Optional[Callable[[Union[bytes, str]], Any]],
       **kwargs
   ) -> Sequence:
     """Opens a reader for a sequence."""
     del kwargs
-    return LineSequence(
-        file_system.open(path, mode), serializer, deserializer
-    )
+    return LineSequence(path, mode, perms, serializer, deserializer)
 

{pyglove-0.4.5.dev202503190809.dist-info → pyglove-0.4.5.dev202503210809.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: pyglove
-Version: 0.4.5.dev202503190809
+Version: 0.4.5.dev202503210809
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors
@@ -31,6 +31,7 @@ Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: keywords
 Dynamic: license
+Dynamic: license-file
 Dynamic: requires-dist
 Dynamic: summary
 

{pyglove-0.4.5.dev202503190809.dist-info → pyglove-0.4.5.dev202503210809.dist-info}/RECORD

@@ -54,9 +54,9 @@ pyglove/core/hyper/numerical_test.py,sha256=UWdH55Bok7bghYDLJOGsgOwV_2LNkhj1AmFw
 pyglove/core/hyper/object_template.py,sha256=YPALTV0mMULa7iuqnryTpA2wMsdyFZ_6g-R525asAr8,22222
 pyglove/core/hyper/object_template_test.py,sha256=TEFX7LIqUvdCdJILnK_gP5xIgNJKzRnioUF0CGVBzcY,9105
 pyglove/core/io/__init__.py,sha256=4ZT1a595DqQuLTNYc2JP_eCp_KesXvHmKRkr777bzpg,785
-pyglove/core/io/file_system.py,sha256=CKb8wpXnoH9VHhY5LzOkUakSdBF3gshLkGxNZBUoZNM,13386
+pyglove/core/io/file_system.py,sha256=E_kSi1Lqo31al4GJYywCzJT97X3ByW8Te4xVfAM93D4,13990
 pyglove/core/io/file_system_test.py,sha256=quJ74i7uEw7aaNIW6steIvn5VKrei5qOLJuOa4Og64w,8692
-pyglove/core/io/sequence.py,sha256=7QWMGXPtJzHyGPgqkT3yJ01FxKJ4mP4lF5HRDiIHNbQ,8165
+pyglove/core/io/sequence.py,sha256=eRxv_LVdsEVgJARiEhDj8m2KNJ2FhPSafW7V00w49OM,8705
 pyglove/core/io/sequence_test.py,sha256=6tmnS7frBuDR8ussT5jugeh23TDsPDrDnbfxHh81Gy4,3891
 pyglove/core/patching/__init__.py,sha256=C1Q1cWPV74YL3eXbzGvc-8aPw1DR8EK6lRhQYDCwHek,2059
 pyglove/core/patching/object_factory.py,sha256=fsiqFET9LWQV4xZXJrPFu3j5IRovSi_zrJba7KiYGZ0,3368
@@ -212,8 +212,8 @@ pyglove/ext/scalars/randoms.py,sha256=LkMIIx7lOq_lvJvVS3BrgWGuWl7Pi91-lA-O8x_gZs
 pyglove/ext/scalars/randoms_test.py,sha256=nEhiqarg8l_5EOucp59CYrpO2uKxS1pe0hmBdZUzRNM,2000
 pyglove/ext/scalars/step_wise.py,sha256=IDw3tuTpv0KVh7AN44W43zqm1-E0HWPUlytWOQC9w3Y,3789
 pyglove/ext/scalars/step_wise_test.py,sha256=TL1vJ19xVx2t5HKuyIzGoogF7N3Rm8YhLE6JF7i0iy8,2540
-pyglove-0.4.5.dev202503190809.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyglove-0.4.5.dev202503190809.dist-info/METADATA,sha256=Ey-JJUgcPxknulbx7rcPNJKfKABWPz7tNhckxQJ_8Z8,7067
-pyglove-0.4.5.dev202503190809.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-pyglove-0.4.5.dev202503190809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
-pyglove-0.4.5.dev202503190809.dist-info/RECORD,,
+pyglove-0.4.5.dev202503210809.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyglove-0.4.5.dev202503210809.dist-info/METADATA,sha256=AGVau_jM2FIr1llnGuNfRwZItvKf97UG3WQ05XfhIGg,7089
+pyglove-0.4.5.dev202503210809.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+pyglove-0.4.5.dev202503210809.dist-info/top_level.txt,sha256=wITzJSKcj8GZUkbq-MvUQnFadkiuAv_qv5qQMw0fIow,8
+pyglove-0.4.5.dev202503210809.dist-info/RECORD,,

{pyglove-0.4.5.dev202503190809.dist-info → pyglove-0.4.5.dev202503210809.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.1.0)
+Generator: setuptools (77.0.3)
 Root-Is-Purelib: true
 Tag: py3-none-any