added piping support

README.md

@@ -1,4 +1,8 @@
-# nma
+# *N*otify *M*e *A*fter
+
+## About
+
+*N*otify *M*e *A*fter (nma) is a python package that sends a desktop notification whenever a shell process or decorated function finishes execution. This repository provides a python shell script for general use and a decorator for use within python code.
 
 ## Requirements:
 
@@ -13,10 +17,6 @@
 
 enter the `nma` directory and run `python3 -m pip install .`
 
-## About
-
-This python package contains a python shell script (and decorator) that sends a desktop notification whenever a shell process or decorated function finishes execution. 
-
 ## Examples
 
 ### In shell scripts
@@ -29,18 +29,14 @@ This python package contains a python shell script (and decorator) that sends a
 
 `nma "sleep 1; echo test"`
 
-or in actual scripts:
+`sleep 1; echo test | nma -`
 
 ```bash
-my_func () {
-    ls *
-    sleep 2
-    echo "nice"   
+my_func() {
+    sleep .1;
 }
-
-nma my_func
+my_func | nma -
 ```
-
 ### Decorator
 
 ```python
@@ -50,3 +46,8 @@ def test():
     print("basic test")
 ```
 
+## Unimplemented Features
+
+* ~~Call script via piping (ex: `sleep 1 | nma -`)~~
+* Email notifications
+* Text notifications

nma/main.py

@@ -12,7 +12,12 @@ def call_from_shell(argv):
     try:
         if len(argv) < 2:
             raise IndexError
-        return subprocess.run(argv[1], shell=True, check=True)
+        if argv[1] == "-":
+            for line in sys.stdin:
+                sys.stdout.write(line)
+            return 0
+        else:
+            return subprocess.run(argv[1], shell=True, check=True)
     except IndexError:
         send_notification("nma error",
                           "shell command not provided", error=True)

nma/notify_decorator.py

@@ -15,11 +15,13 @@ def notify_after(wrapped_func):
     def wrapper(*args, **kwargs):
         res = wrapped_func(*args, **kwargs)
         try:
-            process = "Python"
             name = wrapped_func.__name__
             if name == "call_from_shell":
-                name = process = args[0][1]
-            title = f"{process} execution completed"
+                if args[0][1] == "-":
+                    name = "nma from stdin"
+                else:
+                    name = args[0][1]
+            title = "nma notification"
             message = f"'{name}' has finished executing."
             send_notification(title, message)
         except Exception as e: