rp 0.1.1154__tar.gz → 0.1.1156__tar.gz

{rp-0.1.1154 → rp-0.1.1156}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: rp
-Version: 0.1.1154
+Version: 0.1.1156
 Summary: Ryan's Python
 Home-page: https://github.com/RyannDaGreat/Quick-Python
 Author: Ryan Burgert

{rp-0.1.1154 → rp-0.1.1156}/rp/__main__.py

@@ -17,11 +17,23 @@ def arg_eval(code):
 
 def pop_kwargs():
     kwargs = {}
-    while len(sys.argv) > 3 and sys.argv[-2].startswith('--'):
+    while len(sys.argv) > 3 and rp.starts_with_any(sys.argv[-2], "--", "---"):
         #While we might have a "--kwarg value" at the end
         value = sys.argv.pop()
-        value = arg_eval(value)
-        name  = sys.argv.pop()[len('--'):]
+        name  = sys.argv.pop()
+
+        if name.startswith('---'):
+            #Treated as string
+            name = name[len('---'):]
+
+        elif name.startswith('--'):
+            #Treated as evaluable
+            name = name[len('--'):]
+            value = arg_eval(value)
+
+        else:
+            assert False, 'Sanity check'
+
         kwargs[name] = value
     return kwargs
 
@@ -45,6 +57,9 @@ HELP:
               Format: rp call <funcname> arg1 arg2 arg3 --kwarg1 value1 --kwarg2 value2
       exec    Execute Python code with optional variable assignments
               Format: rp exec <code> --variable1 value1 --variable2 value2
+      
+    Note: All kwargs whose keys start with -- are evaluated as python code, 
+        and all that start with --- are treated as string literals
 
     If no command is provided, an interactive Python terminal (rp.pterm) is started.
 """

{rp-0.1.1154 → rp-0.1.1156}/rp/libs/pynput_recorder.py

@@ -18,7 +18,7 @@ keyboard_controller = KeyboardController()
 class Reprable:
     def __repr__(self):
         attributes = [
-            f"{attr}={repr(value)}"
+            f"{attr}={str(value)}"
             for attr, value in self.__dict__.items()
             if not "__" in attr
         ]
@@ -603,7 +603,7 @@ class PynputCasette(list):
         return PynputCasette(out)
 
     def cap_all_delays(self,seconds):
-        return modify_all_delays(lambda x:min(x,seconds))
+        return self.set_all_delays(lambda x:min(x,seconds))
 
     def play(self, loop=False):
         for _ in range(1 + 999**999 * loop):

{rp-0.1.1154 → rp-0.1.1156}/rp/r.py

@@ -43,6 +43,7 @@ from math import factorial
 # endregion
 # region ［entuple， detuple］
 
+_original_pwd = os.getcwd()
 
 def entuple(x):
     # For pesky petty things.
@@ -5427,7 +5428,7 @@ def save_image_jxl(image, path=None, *, quality=100, add_extension=True):
         ... #png = save_image(final_grid, 'comparison_grid.png')
         
     """
-    pip_import("pillow_jxl", "pillow-jpegxl-plugin")
+    pip_import("pillow_jxl", "pillow-jxl-plugin")
 
     image = as_numpy_image(image)
     if is_grayscale_image(image):
@@ -6512,15 +6513,19 @@ def _display_video_via_mediapy(video, framerate):
 
     return mediapy.show_video(video, fps=framerate)
 
-
-def display_video_in_notebook(video, filetype='mp4', *, framerate=60):
+def display_video_in_notebook(video, filetype='mp4', *, embed=True, framerate=60):
     """
     Display a video or image in a Jupyter notebook.
 
     Args:
         video: The video object to display.
-        filetype (str, optional): The filetype of the video or image. Supported filetypes are 'gif', 'png', 'mp4', and 'avi'. Defaults to 'gif'.
+            - Can be a video: i.e. a list of images as defined by rp.is_image (such as a list of PIL images), or a TCHW torch tensor with values between 0 and 1, or a THWC numpy array
+            - Can be an existing file path: Such as /path/to/video.mp4
+            - Can be a URL: Such as https://file-examples.com/storage/feaef0a3ad67b78fd9cc1df/2017/04/file_example_MP4_480_1_5MG.mp4
+        filetype (str, optional): The filetype of the video or image. Supported filetypes are 'gif', 'png', 'mp4', 'webp' and 'avi'. Defaults to 'gif'.
         framerate (int, optional): The framerate of the video. Defaults to 60.
+        embed (bool, optional): If true, encodes the video as a base-64 string into the notebook itelf - so that the video is saved with the notebook.
+            If false, it will display it as a reference to some file on the host's computer. Good for decreasing filesize of the notebooks.
 
     Raises:
         ValueError: If an unsupported filetype is provided.
@@ -6529,15 +6534,20 @@ def display_video_in_notebook(video, filetype='mp4', *, framerate=60):
         >>> video = create_video(...)
         >>> display_video_in_jupyter_notebook(video, filetype='mp4', embed=True)
     """
-    embed=True # embed (bool, optional): Whether to embed the video or image content directly in the notebook. If False, the video or image is displayed using the file path. Defaults to False.
+    return gather_args_call(_display_video_in_notebook)
+
+def _display_video_in_notebook(video, filetype, *, embed, framerate, save_video=None):
+
+    if save_video is None:
+        save_video=save_video
 
-    from IPython.display import Image, display, HTML
+    from IPython.display import Image, display, HTML, Video
 
     filetype = filetype.strip('.').lower()
-    image_filetypes = 'gif png'.split()
+    image_filetypes = 'gif png webp'.split()
     video_filetypes = 'mp4 avi'.split()
 
-    if filetype=='mp4':
+    if embed and not isinstance(video, str) and filetype=='mp4':
         try:
             return _display_video_via_mediapy(video, framerate)
         except ImportError:
@@ -6546,24 +6556,35 @@ def display_video_in_notebook(video, filetype='mp4', *, framerate=60):
 
     try:
         if isinstance(video, str):
-            assert file_exists(video), 'rp.display_video_in_notebook: Video file {0} does not exist'.format(video)
+            assert is_valid_url(video) or file_exists(video), 'rp.display_video_in_notebook: Video file {0} does not exist'.format(video)
             temp_path = video
+            filetype = get_file_extension(temp_path)
         else:
             temp_path = temporary_file_path(filetype)
             save_video(video, temp_path, framerate=framerate)
 
+        if not embed and filetype in video_filetypes and not is_valid_url(temp_path):
+            #We need a url like http://0.0.0.0:5678/files/TEMP/video.png
+            #For some reason this is needed on videos and NOT images??
+            temp_path = get_relative_path(
+                temp_path,
+                _original_pwd,
+            )
+
         if filetype in image_filetypes:
             if embed:
                 image_hex = file_to_base64(temp_path)
                 display_object = HTML('<img src="data:image/{0};base64,{1}">'.format(filetype, image_hex))
+                # display_object = Image(filename=temp_path, embed=embed) #Equivalent - but the former offers more control if we need it later
             else:
-                display_object = Image(filename=temp_path)
+                assert is_valid_url(temp_path), "I wasn't able to get embed=False to work with image paths yet: temp_path=%s"%temp_path
+                display_object = Image(filename=temp_path, embed=embed)
         elif filetype in video_filetypes:
             if embed:
                 video_hex = file_to_base64(temp_path)
                 display_object = HTML('<video loop autoplay controls><source src="data:video/{0};base64,{1}" type="video/{0}"></video>'.format(filetype, video_hex))
             else:
-                display_object = HTML('<video loop autoplay controls><source src="{0}" type="video/{1}"></video>'.format(temp_path, filetype))
+                display_object = Video(filename=temp_path, html_attributes='autoplay loop controls')
         else:
             raise ValueError("rp.display_video_in_notebook: Unsupported filetype: {0}. Supported filetypes are {1}".format(filetype, ', '.join(image_filetypes + video_filetypes)))
 
@@ -6573,6 +6594,25 @@ def display_video_in_notebook(video, filetype='mp4', *, framerate=60):
         if not isinstance(video, str) and embed and file_exists(temp_path):
             delete_file(temp_path)
 
+def display_video_in_notebook_webp(video, quality=100, framerate=60, embed=False):
+    """
+    Displays an animated webp in a Jupyter notebook with a specified quality and framerate
+    See rp.display_video_in_notebook's docstring for explanations of what the args do
+
+    EXAMPLE:
+        >>> import rp
+        ... video_url = "https://file-examples.com/storage/feaef0a3ad67b78fd9cc1df/2017/04/file_example_MP4_480_1_5MG.mp4"
+        ... video = rp.load_video(video_url, use_cache=True)
+        ... 
+        ... for quality in [1, 10, 25, 50, 90, 95, 100]:
+        ...     #Ranges from ~500KB to 14MB
+        ...     print(quality)
+        ...     rp.display_video_in_notebook_webp(video, quality)
+    """
+    def save_video(video, path, framerate):
+        return save_video_webp(video, path, quality=quality, framerate=framerate)
+    return gather_args_call(_display_video_in_notebook, filetype='webp')
+
 # def display_embedded_video_in_notebook(video,framerate:int=30,filetype:str='gif'):
 #     """
 #     This will embed a video into the jupyter notebook you're using
@@ -10003,6 +10043,17 @@ def rinsp(object,search_or_show_documentation:bool=False,show_source_code:bool=F
     if isinstance(object, int) and process_exists(object):
         print_process_info(object)
 
+
+    if isinstance(object, int) and get_port_is_taken(object):
+        try:
+            process=get_process_using_port(object)
+            by_string = 'BY PROCESS %i'%(process)
+            print(col(tab + 'PORT %i IS TAKEN '%object + by_string))
+            print_process_info(process)
+        except Exception as e:
+            by_string = '(unknown process: %s)'%repr(e)
+            print(col(tab + 'PORT %i IS TAKEN '%object + by_string))
+                
  
     if is_symlink(object):
         print(col(tab + 'SYMLINK --> '+read_symlink(object)))
@@ -10668,6 +10719,19 @@ def append_line_to_file(line:str,file_path:str):
     if not file_exists(file_path):
         string_to_text_file(file_path,line)
     else:
+
+
+
+
+
+
+
+
+
+
+
+
+
         file=open(file_path, 'a')
         try:
             file.write('\n'+line)
@@ -18197,6 +18261,7 @@ def pseudo_terminal(
         DQ CDHQ FAST
 
         PRP PYM rp
+        SURP $os.system('sudo '+sys.executable+' -m rp')
 
         GOO  $open_google_search_in_web_browser(str(ans))
         GOOP $open_google_search_in_web_browser($string_from_clipboard())
@@ -21235,6 +21300,8 @@ def save_animated_png(frames, path=None,*,framerate=None):
     make_parent_directory(path)
     save_animated_png(path, frames, delay=delay)
 
+    path = get_absolute_path(path)
+
     return path
 
 save_video_png = save_animated_png
@@ -27760,7 +27827,10 @@ def _images_conversion(func, images, *, copy_check ,copy=True):
             assert len(images.shape)==4, images.shape #BHWC
             C=images.shape[3]
             if C==3:
-                return np.concatenate(images,np.ones_like(images[:,:,:1]))
+                alpha=np.ones_like(images[:,:,:,:1])
+                if images.dtype==np.uint8:
+                    alpha*=255
+                return np.concatenate((images,alpha),3)
             else:
                 assert C==4, C
                 return images.copy()
@@ -29584,6 +29654,8 @@ def save_video_avi(frames,path:str=None,framerate:int=30):
         frame=as_byte_image(frame)
         frame=cv_bgr_rgb_swap(frame)
         out.write(frame)
+
+    path = rp.get_absolute_path(path)
         
     return path
 
@@ -29921,6 +29993,8 @@ def save_video_gif_via_pil(video, path=None, *, framerate=30):
         loop=0,
     )
 
+    path = get_absolute_path(path)
+
     return path
 
 save_animated_gif = save_video_gif = save_animated_gif_via_pil = save_video_gif_via_pil
@@ -30112,7 +30186,7 @@ def save_video(images, path, *, framerate=60):
     """
     assert not isinstance(images, str), 'The first argument should be the sequence of video frames, not the path!'
 
-    assert get_file_extension(path) in 'mp4 avi gif png'.split(), 'This function currently supports .mp4, .avi, .png and .gif files'
+    assert get_file_extension(path) in 'mp4 avi gif png webp'.split(), 'This function currently supports .mp4, .avi, .png, .webp and .gif files'
 
     if path.endswith('.mp4') : return save_video_mp4(images,path,framerate=framerate, show_progress=False)
     if path.endswith('.avi') : return save_video_avi(images,path,framerate=framerate)
@@ -35088,6 +35162,97 @@ def get_next_free_port(port):
         port+=1
     return port
 
+_get_all_taken_ports_cache = None
+
+
+def get_all_taken_ports(
+    *, lazy=False, num_threads=None, show_progress=False, use_cache=False
+):
+    """Returns all ports that are currently taken up"""
+    global _get_all_taken_ports_cache
+    if use_cache and _get_all_taken_ports_cache is not None:
+        return _get_all_taken_ports_cache
+
+    def helper(x):
+        assert get_port_is_taken(x)
+        return x
+
+    if show_progress is True:
+        show_progress = "eta:" + get_current_function_name()
+    output = load_files(
+        helper,
+        range(65536),
+        strict=False,
+        lazy=lazy,
+        num_threads=None,
+        show_progress=show_progress,
+    )
+    if not lazy:
+        _get_all_taken_ports_cache = output
+    return output
+
+def get_process_using_port(port: int, *, strict = True):
+    """
+    Gets the process ID (PID) using the specified port.
+
+    Args:
+        port: The port number.
+        strict: If True, raise RuntimeError if no process is found.
+                If None, return None if no process is found.
+
+    Returns:
+        The PID, or None if strict is None and no process is found.
+
+    Raises:
+        TypeError: If 'port' is not an int, or 'strict' is not bool or None.
+        ValueError: If 'port' is out of range.
+        RuntimeError: If strict is True and no process/PID is found.
+        psutil.Error: If there's a psutil error (rare).
+
+    EXAMPLES:
+        >>> # Assuming a process is listening on port 8080
+        >>> port = 8080
+        >>> assert get_port_is_taken(port)
+        >>> pid = get_process_using_port(port)
+        >>> print(pid)  # Output would be the PID (e.g., 1234)
+
+        >>> # If no process is listening on port 9999, with strict=True (default)
+        >>> port = 9999
+        >>> try:
+        ...   get_process_using_port(port)
+        ... except RuntimeError as e:
+        ...   print(str(e))
+        No process found using port 9999.
+
+        >>> # If no process is listening on port 9999, with strict=None
+        >>> port = 9999
+        >>> result = get_process_using_port(port, strict=None)
+        >>> print(result)
+        None
+    """
+    pip_import('psutil')
+    import psutil
+
+    if not isinstance(port, int):
+        raise TypeError("Port must be an integer but got type %s"%type(port))
+    if not 0 <= port <= 65535:
+        raise ValueError("Port must be within the range 0-65535 but got %i"%port)
+    if strict not in [True, None]:
+        raise TypeError("'strict' argument must be either True or None but got %s"%strict)
+
+    for conn in psutil.net_connections():
+        if conn.laddr.port == port and conn.status == psutil.CONN_LISTEN:
+            if conn.pid:
+                return conn.pid
+            elif strict: # Handle edge case where PID might not be available
+                raise RuntimeError("Process found using port {0}, but PID is unavailable.".format(port))
+            else:
+                return None
+
+    if strict:
+        raise RuntimeError("No process found using port {0}.".format(port))
+    else:
+        return None
 
 
 def can_convert_object_to_bytes(x:object)->bool:
@@ -36993,11 +37158,16 @@ def bytes_to_file(data: bytes, path: str = None):
 
 _file_to_bytes_cache={}
 def file_to_bytes(path: str, use_cache=False):
+
     if use_cache and path in _file_to_bytes_cache:
         return _file_to_bytes_cache[path]
 
-    with open(path, 'rb') as out:
-        data = out.read()
+    if is_valid_url(path):
+        data = curl_bytes(path)
+
+    else:
+        with open(path, 'rb') as out:
+            data = out.read()
 
     if use_cache:
         _file_to_bytes_cache[path]=_file_to_bytes_cache

{rp-0.1.1154 → rp-0.1.1156}/rp/rp_ptpython/key_bindings.py

@@ -2362,7 +2362,7 @@ def load_python_bindings(python_input):
     #     buffer.insert_text(")")
     #     buffer.cursor_left(count=1)
 #endregion
-    for char in '''`~!@#$%^&*()-_=+[{]}\|;:'",<.>/?']''':
+    for char in r'''`~!@#$%^&*()-_=+[{]}\|;:'",<.>/?']''':
         def go(c):
             @handle(c,filter=~vi_mode_enabled&microcompletions_enabled)
             def _(event):
@@ -2715,6 +2715,15 @@ def load_python_bindings(python_input):
                                          '\\tbp':'toggle_big_parenthesis',
                                          '\\spa':'string_paste',
                                          '\\d0l':'delete_empty_lines',
+                                         '\\d1l':'delete_empty_lines_1',
+                                         '\\d2l':'delete_empty_lines_2',
+                                         '\\d3l':'delete_empty_lines_3',
+                                         '\\d4l':'delete_empty_lines_4',
+                                         '\\d5l':'delete_empty_lines_5',
+                                         '\\d6l':'delete_empty_lines_6',
+                                         '\\d7l':'delete_empty_lines_7',
+                                         '\\d8l':'delete_empty_lines_8',
+                                         '\\d9l':'delete_empty_lines_9',
                                          '\\dtt':'delete_to_top',
                                          '\\dtb':'delete_to_bottom',
                                          '\\sl':'sort_lines',
@@ -3013,6 +3022,20 @@ def load_python_bindings(python_input):
                                     text=buffer.document.text
                                     text='\n'.join(line for line in text.split('\n') if line.strip())
                                     buffer.document=Document(text,min(len(text),buffer.document.cursor_position),buffer.document.selection)
+                                if header.startswith('delete_empty_lines_'):
+                                    N=header[len('delete_empty_lines_'):]
+                                    try:
+                                        N=int(N)
+                                        N+=1
+                                        import rp
+                                        text=buffer.document.text
+                                        text=rp.strip_trailing_whitespace(text)
+                                        while '\n'*(N+1) in text:
+                                            text=text.replace('\n'*(N+1),'\n'*N)
+                                        buffer.document=Document(text,min(len(text),buffer.document.cursor_position),buffer.document.selection)
+                                    except e:
+                                        buffer.insert_text(str(e))
+
                                 if header=='sort_lines':
                                     text=buffer.document.text
                                     text='\n'.join(sorted(text.split('\n')))

rp-0.1.1156/rp/version.py

@@ -0,0 +1 @@
+1156

{rp-0.1.1154 → rp-0.1.1156}/rp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: rp
-Version: 0.1.1154
+Version: 0.1.1156
 Summary: Ryan's Python
 Home-page: https://github.com/RyannDaGreat/Quick-Python
 Author: Ryan Burgert

rp-0.1.1156/version.py

@@ -0,0 +1 @@
+1156

rp-0.1.1154/rp/version.py

@@ -1 +0,0 @@
-1154

rp-0.1.1154/version.py

@@ -1 +0,0 @@
-1154