gmsg 0.1.0__py3-none-any.whl

gmsg/__init__.py

@@ -0,0 +1 @@
+from .gmsg import main

gmsg/api_key.py

@@ -0,0 +1,20 @@
+import sys
+import os
+
+from pathlib import Path
+
+
+def get_or_set_api_key():
+    try:
+        rc = os.path.join(Path.home(), ".gmsgrc")
+        if not os.path.isfile(rc):
+            key_inp = input("Enter your Gemini API KEY: ").strip()
+            with open(rc, 'w') as f:
+                f.write(key_inp)
+
+            return key_inp
+
+        with open(rc, 'r') as f:
+            return f.read().strip()
+    except KeyboardInterrupt:
+        sys.exit(0)

gmsg/gmsg.py

@@ -0,0 +1,104 @@
+import sys
+import subprocess
+from google import genai
+from pathlib import Path
+from .api_key import get_or_set_api_key
+
+
+def is_git_repo() -> bool:
+    path = Path(Path.cwd())
+    for parent in [path] + list(path.parents):
+        if (parent / ".git").exists():
+            return True
+    return False
+
+
+def git_diff() -> str | None:
+    try:
+        result = subprocess.run(["git", "diff", "--cached"],
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                check=True,
+                                text=True)
+
+        if not result.stdout:
+            return None
+        return result.stdout
+    except subprocess.CalledProcessError:
+        sys.exit(1)
+
+
+def make_query(diff) -> str:
+    prompt = f"Generate a one liner git commit message for these changes, Below are the changes:\n {diff}"
+    return prompt
+
+
+def generate_commit_message(diff: str) -> str:
+    try:
+        query = make_query(diff)
+        while True:
+            client = genai.Client(api_key=USER_API_KEY)
+            msg = client.models.generate_content(model="gemini-2.0-flash",
+                                                 contents=query)
+
+            print(msg.text)
+            # generated_msg = "feat: Add poetry.lock and pyproject.toml files."
+            # print(generated_msg)
+            if continue_with_this_prompt(msg.text):
+                break
+    except genai.errors.ClientError as error:
+        if error.code == 400:
+            print(dict(error="Gemini API key not valid, check ~/.gmsg"))
+        else:
+            print(error)
+            sys.exit(1)
+
+
+def continue_with_this_prompt(commit_msg: str) -> bool:
+    action = input("Do you want to continue with this message [Y/n]: ")
+    if action in ("", "y", "Y"):
+        print(f"Running: `git commit -m {commit_msg}`")
+        print(
+            f"Message commited to git, to confirm you can run git commit --amend."
+        )
+        commit_message_to_git(commit_msg)
+        return True
+    elif action in ("n", "N"):
+        return False
+
+
+def commit_message_to_git(generated_msg: str) -> str | None:
+    try:
+        result = subprocess.run(["git", "commit", "-m", generated_msg],
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                check=True,
+                                text=True)
+
+        return result.stdout
+    except subprocess.CalledProcessError:
+        return None
+
+
+def main():
+    global USER_API_KEY
+
+    USER_API_KEY = get_or_set_api_key()
+    try:
+        if not is_git_repo():
+            print("Current directory does not have git initialized",
+                  file=sys.stdout)
+            sys.exit(1)
+        diff = git_diff()
+        if not diff:
+            print("There's nothing in git diff.")
+            sys.exit(1)
+
+        generate_commit_message(diff)
+
+    except KeyboardInterrupt:
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()

gmsg-0.1.0.dist-info/METADATA

@@ -0,0 +1,15 @@
+Metadata-Version: 2.3
+Name: gmsg
+Version: 0.1.0
+Summary: Generates git commit messages for you using AI.
+License: MIT
+Author: shiyan99s@gmail.com
+Requires-Python: >=3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: google-genai (>=1.9.0,<2.0.0)
+Description-Content-Type: text/markdown
+
+

gmsg-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+gmsg/__init__.py,sha256=9B4ZkbOBiBYYqv3Kh-mdlNn-yDPvkpXOqrm52Ng8CTk,23
+gmsg/api_key.py,sha256=92X--Gmwki2QxBSpMwo5xTl6I1g6GMw8FbCwVx8ojME,454
+gmsg/gmsg.py,sha256=NmQd55li6D5g8vR2hY1gxKXarm0XPSG9jyPO63RvS5g,3016
+gmsg-0.1.0.dist-info/METADATA,sha256=GusjYVbsPUYN-LV7Mn7lg30VvidTTJjFpmUdjfZCS94,458
+gmsg-0.1.0.dist-info/WHEEL,sha256=RaoafKOydTQ7I_I3JTrPCg6kUmTgtm4BornzOqyEfJ8,88
+gmsg-0.1.0.dist-info/entry_points.txt,sha256=LqLzIXT9IVenl_xEptCn4LqWHm34ypTaVMhoW5tFBSg,39
+gmsg-0.1.0.dist-info/RECORD,,

gmsg-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.0.0
+Root-Is-Purelib: true
+Tag: py3-none-any

gmsg-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+gmsg=gmsg.gmsg:main
+