ipyvasp 1.0.9__py2.py3-none-any.whl → 1.1.0__py2.py3-none-any.whl

ipyvasp/_lattice.py

@@ -231,6 +231,21 @@ def periodic_table(selection=None):
     ax.set(xlim=[-0.6,17.6],ylim=[9.6,-0.6]) # to show borders correctly
     return ax
 
+def _write_text(dest, text: str, *, encoding: str = "utf-8") -> None:
+    "Write unicode text either to a path-like destination or to a writable text stream."
+    # Treat file-like objects (streams) first (avoid Path("CON") / weird Windows devices, etc.)
+    if hasattr(dest, "write") and callable(getattr(dest, "write")):
+        dest.write(text)
+        # Best-effort flush (sys.stdout has it, StringIO doesn't need it)
+        flush = getattr(dest, "flush", None)
+        if callable(flush):
+            flush()
+        return
+
+    # Otherwise treat as a filesystem path
+    path = Path(dest)
+    with path.open("w", encoding=encoding) as f:
+        f.write(text)
 
 def write_poscar(poscar_data, outfile=None, selective_dynamics=None, overwrite=False, comment="", scale=None, system=None):
     """Writes POSCAR data to a file or returns string
@@ -285,19 +300,18 @@ def write_poscar(poscar_data, outfile=None, selective_dynamics=None, overwrite=F
         pos_list = [f"{p}   {s}" for p, s in zip(pos_list, sd)]
 
     out_str += "\n".join(pos_list)
-    if outfile:
-        path = Path(outfile)
-        if not path.is_file():
-            with path.open("w", encoding="utf-8") as f:
-                f.write(out_str)
-
-        elif overwrite and path.is_file():
-            with path.open("w", encoding="utf-8") as f:
-                f.write(out_str)
+    if outfile is not None:
+        # If it's a writable stream (sys.stdout, StringIO, open file handle), write directly.
+        if hasattr(outfile, "write") and callable(getattr(outfile, "write")):
+            _write_text(outfile, out_str)
         else:
-            raise FileExistsError(
-                f"{outfile!r} exists, can not overwrite, \nuse overwrite=True if you want to change."
-            )
+            # Otherwise treat as path-like with overwrite protection.
+            path = Path(outfile)
+            if path.exists() and not overwrite:
+                raise FileExistsError(
+                    f"{str(path)!r} exists, can not overwrite; use overwrite=True."
+                )
+            _write_text(path, out_str)
     else:
         print(out_str)
 
@@ -555,9 +569,8 @@ class InvokeMaterialsProject:
                 return f"Structure(unit={self.unit},mp_id={self.mp_id!r},symbol={self.symbol!r},crystal={self.crystal!r},cif='{self._cif[:10]}...')"
 
             def write_cif(self, outfile=None):
-                if isinstance(outfile, str):
-                    with open(outfile, "w") as f:
-                        f.write(self._cif)
+                if outfile is not None:
+                    _write_text(outfile, self._cif)
                 else:
                     print(self._cif)
 
@@ -640,12 +653,12 @@ def get_kpath(
     ibzkpt : PathLike
         Path to ibzkpt file, required for HSE calculations.
     outfile : PathLike
-        Path/to/file to write kpoints.
+        Path/to/file to write kpoints. Use sys.stdout to print to console.
     rec_basis : array_like
         Reciprocal basis 3x3 array to use for calculating uniform points.
 
 
-    If `outfile = None`, a tuple of header and kpoints array (Nx3) is returned.
+    If `outfile = None`, kpoints array (Nx3) is returned.
     """
     if isinstance(kpoints, str):
         kpoints = _str2kpoints(kpoints)
@@ -766,10 +779,9 @@ def get_kpath(
     )
     out_str = "{}\n{}".format(top_str, out_str)
     if outfile != None:
-        with open(outfile, "w", encoding="utf-8") as f:  # allow unicode
-            f.write(out_str)
+        _write_text(outfile, out_str)
     else:
-        return top_str, points  # return points as well for any processing by user.
+        return points  # return points for any processing by user.
 
 
 # Cell
@@ -799,12 +811,12 @@ def get_kmesh(
     ibzkpt : PathLike
         Path to ibzkpt file, required for HSE calculations.
     outfile : PathLike
-        Path/to/file to write kpoints.
+        Path/to/file to write kpoints. Use sys.stdout to print to console.
     endpoint : bool
         Default True, include endpoints in mesh at edges away from origin.
 
 
-    If `outfile = None`, a tuple of header and kpoints array (Nx3) is returned.
+    If `outfile = None`, kpoints array (Nx3) is returned.
 
     """
     if len(args) not in [1, 3]:
@@ -879,10 +891,9 @@ def get_kmesh(
     )
     out_str = "{}\n{}".format(top_str, out_str)
     if outfile != None:
-        with open(outfile, "w", encoding="utf-8") as f:
-            f.write(out_str)
+        _write_text(outfile, out_str)
     else:
-        return top_str, points # return points as well for any processing by user.
+        return points # return points for any processing by user.
 
 
 # Cell

ipyvasp/_version.py

@@ -1 +1 @@
-__version__ = "1.0.9"
+__version__ = "1.1.0"

{ipyvasp-1.0.9.dist-info → ipyvasp-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ipyvasp
-Version: 1.0.9
+Version: 1.1.0
 Summary: A processing tool for VASP DFT input/output processing in Jupyter Notebook.
 Home-page: https://github.com/massgh/ipyvasp
 Author: Abdul Saboor

{ipyvasp-1.0.9.dist-info → ipyvasp-1.1.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 ipyvasp/__init__.py,sha256=pzTqeKuf6sN2GQmaexmMgG677ggT3sxIFyQDXq_2whU,1422
 ipyvasp/__main__.py,sha256=eJV1TZSiT8mC_VqAeksNnBI2I8mKMiPkEIlwikbtOjI,216
 ipyvasp/_enplots.py,sha256=gJ7S9WBmrxvDEbmoccDRaJG01kpx9oNlRf7mozigbgY,37872
-ipyvasp/_lattice.py,sha256=6DRC_GZlsmdxn94ktULVCZDRHZKJwm5iZ6kEqg9Psk0,106579
-ipyvasp/_version.py,sha256=5fdKqtSBBDvdwuTWCGoh62x6-wR269e8DEQnOPkCHWg,23
+ipyvasp/_lattice.py,sha256=-JQaIdB2e6yFh5j_U2343FH2aM_c3cSdqSC19cJINq4,107218
+ipyvasp/_version.py,sha256=Zrv57EzpjdsuSPqsYvFkVsQKKRUOHFG7yURCf7qN-Tk,23
 ipyvasp/bsdos.py,sha256=hVHpxkdT2ImRsxwFvMSMHxRSo4LqDM90DnUhwTP8vcs,32192
 ipyvasp/cli.py,sha256=-Lf-qdTvs7WyrA4ALNLaoqxMjLsZkXdPviyQps3ezqg,6880
 ipyvasp/evals_dataframe.py,sha256=n2iSH4D4ZbrxlAV4yDTVHbcl3ycfD0zfQYmTBcxjfkE,20789
@@ -17,9 +17,9 @@ ipyvasp/core/parser.py,sha256=o-uHyL_w9W0pmxoSt3JmLwwmmT3WRHlHSs_NoiHO-hs,39401
 ipyvasp/core/plot_toolkit.py,sha256=aURiPAu0tWTNctuA9LMEL3CfraRgAOyTJURzDrrO0r0,36232
 ipyvasp/core/serializer.py,sha256=aEc7K5jVga8gxm9Tt2OgBw8wnsmWZGtODBnwRJ_5sf0,38423
 ipyvasp/core/spatial_toolkit.py,sha256=dXowREhiFzBvvr5f_bApzFhf8IzjH2E2Ix90oCBUetY,14885
-ipyvasp-1.0.9.dist-info/LICENSE,sha256=F3SO5RiAZOMfmMGf1KOuk2g_c4ObvuBJhd9iBLDgXoQ,1263
-ipyvasp-1.0.9.dist-info/METADATA,sha256=E5y8hHvowwd5WtJRtlSotIgjLCkNEXp9DL7fgrHNCM4,3218
-ipyvasp-1.0.9.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
-ipyvasp-1.0.9.dist-info/entry_points.txt,sha256=aU-gGjQG2Q8XfxDlNc_8__cwfp8WG2K5ZgFPInTm2jg,45
-ipyvasp-1.0.9.dist-info/top_level.txt,sha256=ftziWlMWu_1VpDP1sRTFrkfBnWxAi393HYDVu4wRhUk,8
-ipyvasp-1.0.9.dist-info/RECORD,,
+ipyvasp-1.1.0.dist-info/LICENSE,sha256=F3SO5RiAZOMfmMGf1KOuk2g_c4ObvuBJhd9iBLDgXoQ,1263
+ipyvasp-1.1.0.dist-info/METADATA,sha256=DCZLpgRjMRdVVJqT23OoP4AVFf_HeLUICtdBdSYa4_I,3218
+ipyvasp-1.1.0.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
+ipyvasp-1.1.0.dist-info/entry_points.txt,sha256=aU-gGjQG2Q8XfxDlNc_8__cwfp8WG2K5ZgFPInTm2jg,45
+ipyvasp-1.1.0.dist-info/top_level.txt,sha256=ftziWlMWu_1VpDP1sRTFrkfBnWxAi393HYDVu4wRhUk,8
+ipyvasp-1.1.0.dist-info/RECORD,,