M-Porter · October 1, 2021 21:57 · Oct 1, 2021 · Oct 1, 2021 · Oct 1, 2021
diff --git a/git-pt b/git-pt
@@ -19,14 +19,12 @@ class PullThis:
         except IndexError:
             return "origin"
 
-
     def __get_branch(self) -> str:
         branch = subprocess.check_output(
             ["git", "rev-parse", "--abbrev-ref", "head"]
         ).decode("utf-8")
         return branch.strip()
 
-
     def run(self):
         """Pull down the latest commits from the active working branch.
 

diff --git a/git-pt b/git-pt
@@ -5,11 +5,11 @@ import subprocess
 
 
 class Colors:
-    CYAN = '\033[96m'
-    GREEN = '\033[92m'
-    WARNING = '\033[93m'
-    FAIL = '\033[91m'
-    ENDC = '\033[0m'
+    CYAN = "\033[96m"
+    GREEN = "\033[92m"
+    WARNING = "\033[93m"
+    FAIL = "\033[91m"
+    ENDC = "\033[0m"
 
 
 class PullThis:

diff --git a/git-pt b/git-pt
@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+
+import sys
+import subprocess
+
+
+class Colors:
+    CYAN = '\033[96m'
+    GREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    ENDC = '\033[0m'
+
+
+class PullThis:
+    def __get_origin(self) -> str:
+        try:
+            return sys.argv[1]
+        except IndexError:
+            return "origin"
+
+
+    def __get_branch(self) -> str:
+        branch = subprocess.check_output(
+            ["git", "rev-parse", "--abbrev-ref", "head"]
+        ).decode("utf-8")
+        return branch.strip()
+
+
+    def run(self):
+        """Pull down the latest commits from the active working branch.
+
+        >>> git pt
+
+        Optionally, you can pull down commits from another remote on the 
+        current branch by providing an argument. By default, the command will 
+        always pull from origin.
+
+        >>> git pt another-remote
+        """
+        origin = self.__get_origin()
+
+        try:
+            branch = self.__get_branch()
+        except subprocess.CalledProcessError:
+            exit(1)
+
+        print(f"{Colors.GREEN}Pulling down latest...{Colors.ENDC}")
+        print(f"{Colors.CYAN}=>{Colors.ENDC} git pull {origin} {branch}\n")
+
+        subprocess.run(["git", "pull", origin, branch])
+
+
+if __name__ == "__main__":
+    PullThis().run()
No results found