pyglove 0.4.5.dev202503200809__tar.gz → 0.4.5.dev202503220808__tar.gz

{pyglove-0.4.5.dev202503200809 → pyglove-0.4.5.dev202503220808}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyglove
-Version: 0.4.5.dev202503200809
+Version: 0.4.5.dev202503220808
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors

{pyglove-0.4.5.dev202503200809 → pyglove-0.4.5.dev202503220808}/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-0.4.5.dev202503200809 → pyglove-0.4.5.dev202503220808}/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.dev202503200809 → pyglove-0.4.5.dev202503220808}/pyglove.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyglove
-Version: 0.4.5.dev202503200809
+Version: 0.4.5.dev202503220808
 Summary: PyGlove: A library for manipulating Python objects.
 Home-page: https://github.com/google/pyglove
 Author: PyGlove Authors