kbasic 0.1.7__tar.gz → 0.1.8__tar.gz

{kbasic-0.1.7 → kbasic-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kbasic
-Version: 0.1.7
+Version: 0.1.8
 Summary: Keyan's basic utility functions.
 Requires-Dist: numpy>=2.4.2
 Requires-Dist: pylatexenc>=2.10

{kbasic-0.1.7 → kbasic-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kbasic"
-version = "0.1.7"
+version = "0.1.8"
 description = "Keyan's basic utility functions."
 readme = "README.md"
 requires-python = ">=3.11"

{kbasic-0.1.7 → kbasic-0.1.8}/src/kbasic/__init__.py

@@ -1,3 +1,4 @@
+from kbasic.environment import *
 from kbasic.audio import *
 from kbasic.array import *
 from kbasic.bar import *

kbasic-0.1.8/src/kbasic/environment/Keyan.py

@@ -0,0 +1,12 @@
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                             Imports                             <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+from kbasic.parsing import Folder
+
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                           Definitions                           <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+simulationDir = Folder("/Users/keyan/code/data/sims/")
+figDir = videoDir = Folder("./")
+frameDir = Folder("/Users/keyan/code/data/frames/")
+videoDir = Folder("/Users/keyan/videos/")

kbasic-0.1.8/src/kbasic/environment/__init__.py

@@ -0,0 +1,24 @@
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                             Imports                             <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+from kbasic.parsing import File, Folder
+
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                           Definitions                           <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+isAnvil: bool = Folder("/anvil").exists
+isPC: bool = Folder("/Users").exists
+isWindows: bool = Folder("/C:").exists
+if isAnvil: 
+    from kgsim.environment.anvil import simulationDir # so shit don't complain
+    from kgsim.environment.anvil import *
+elif isPC:
+    user = File(__file__).path.lower().split('/')[2]
+    match user:
+        case 'keyan': from kgsim.environment.Keyan import *
+        case _: from kgsim.environment.defaultPC import *
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                            Functions                            <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+def available() -> list[str]: 
+    return simulationDir.children

kbasic-0.1.8/src/kbasic/environment/anvil.py

@@ -0,0 +1,13 @@
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                             Imports                             <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+from kbasic.parsing import Folder
+
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                           Definitions                           <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+anvil_user = __file__.split("/")[2]
+simulationDir = Folder(f"/anvil/scratch/{anvil_user}/sims/") # default location to put/look for simulations
+figDir = Folder(f"/home/{anvil_user}/turbulence/figures/") # default location for figures to go
+frameDir = Folder(f"/home/{anvil_user}/frames/") # default location for frames of videos to go
+videoDir = Folder(f"/home/{anvil_user}/videos/") # default location for videos to go

kbasic-0.1.8/src/kbasic/environment/defaultPC.py

@@ -0,0 +1,13 @@
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                             Imports                             <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+from kbasic.parsing import Folder
+
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+# >-|===|>                           Definitions                           <|===|-<
+# !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+user = __file__.split("/")[1]
+simulationDir = Folder(f"/Users/{user}/sims/")
+figDir = Folder(f"/Users/{user}/figures/")
+frameDir = Folder(f"/Users/{user}/frames/")
+videoDir = Folder(f"/Users/{user}/videos/")

{kbasic-0.1.7 → kbasic-0.1.8}/src/kbasic/shell.py

@@ -3,6 +3,8 @@
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 from kbasic.bar import ProgressBar, redirect_to_tqdm
 from kbasic.audio import success
+from kbasic.environment import isAnvil 
+if isAnvil: from kbasic.environment.anvil import anvil_user
 # from kgsim.dhybridr.io import dHybridRinput
 from subprocess import check_output, DEVNULL
 from tqdm import tqdm
@@ -16,12 +18,24 @@ import numpy as np
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 # >-|===|>                           Definitions                           <|===|-<
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
-_USERNAME_ = 'x-kgootkin'
 bad = ['\x1b[31m', '\x1b[34m', '\x1b[m']
-
+ansi = {"BLACK": u"\033[0;30m","RED": u"\x1b[0;31m","GREEN": u"\033[0;32m","BROWN": u"\033[0;33m","BLUE": u"\033[0;34m","PURPLE": u"\033[0;35m","CYAN": u"\033[0;36m","WHITE": u"\033[0;37m",
+    "FAINT": u"\033[2m","FAINT BLACK": u"\033[2;30m","FAINT RED": u"\x1b[2;31m","FAINT GREEN": u"\033[2;32m","FAINT BROWN": u"\033[2;33m","FAINT BLUE": u"\033[2;34m","FAINT PURPLE": u"\033[2;35m","FAINT CYAN": u"\033[2;36m","FAINT WHITE": u"\033[2;37m",
+    "ITALIC": u"\033[3m","ITALIC BLACK": u"\033[3;30m","ITALIC RED": u"\x1b[3;31m","ITALIC GREEN": u"\033[3;32m","ITALIC BROWN": u"\033[3;33m","ITALIC BLUE": u"\033[3;34m","ITALIC PURPLE": u"\033[3;35m","ITALIC CYAN": u"\033[3;36m","ITALIC WHITE": u"\033[3;37m",
+    "UNDERLINE": u"\033[4m","UNDERLINE BLACK": u"\033[4;30m","UNDERLINE RED": u"\x1b[4;31m","UNDERLINE GREEN": u"\033[4;32m","UNDERLINE BROWN": u"\033[4;33m","UNDERLINE BLUE": u"\033[4;34m","UNDERLINE PURPLE": u"\033[4;35m","UNDERLINE CYAN": u"\033[4;36m","UNDERLINE WHITE": u"\033[4;37m",
+    "BLINK": u"\033[5m","BLINK BLACK": u"\033[5;30m","BLINK RED": u"\x1b[5;31m","BLINK GREEN": u"\033[5;32m","BLINK BROWN": u"\033[5;33m","BLINK BLUE": u"\033[5;34m","BLINK PURPLE": u"\033[5;35m","BLINK CYAN": u"\033[5;36m","BLINK WHITE": u"\033[5;37m",
+    "NEGATIVE": u"\033[7m","BLACK BACKGROUND": u"\033[7;30m","RED BACKGROUND": u"\x1b[7;31m","GREEN BACKGROUND": u"\033[7;32m","BROWN BACKGROUND": u"\033[7;33m","BLUE BACKGROUND": u"\033[7;34m","PURPLE BACKGROUND": u"\033[7;35m","CYAN BACKGROUND": u"\033[7;36m","WHITE BACKGROUND": u"\033[7;37m",
+    "END": u"\033[0m"
+}
+_USERNAME_ = None if not isAnvil else anvil_user
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 # >-|===|>                            Functions                            <|===|-<
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
+def pprint(text: str, option: str = 'white'):
+    assert option.upper() in ansi.keys(), f"must give one of {ansi.keys()} as your option. "
+    output = ansi[option.upper()] + text + ansi['END']
+    print(output)
+def warn(text: str): pprint(text, 'yellow')
 def parse_shell_output(output):
         match output:
             case str(): return output 
@@ -59,19 +73,6 @@ qs = anvil_queue
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==
 # >-|===|>                             Classes                             <|===|-<
 # !==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!==!== 
-class Color:
-    HEADER = '\033[95m'
-    OKBLUE = '\033[94m'
-    OKCYAN = '\033[96m'
-    OKGREEN = '\033[92m'
-    WARNING = '\033[93m'
-    FAIL = '\033[91m'
-    ENDC = '\033[0m'
-    BOLD = '\033[1m'
-    UNDERLINE = '\033[4m'
-    def warn(self, text: str):
-        print(self.WARNING + text + self.ENC)
-
 class AnvilJob:
     def __init__(self, queue_row: str, sep="DISTINCTSEPERATOR"):
         self.sep = sep