add copy to clipboard

code2prompt/write_output.py

@@ -1,12 +1,14 @@
 import click
 from pathlib import Path
+import pyperclip
 
 def write_output(content, output_path):
     """
-    Writes the generated content to a file or prints it to the console.
+    Writes the generated content to a file or prints it to the console,
+    and copies the content to the clipboard.
 
     Parameters:
-    - content (str): The content to be written or printed.
+    - content (str): The content to be written, printed, and copied.
     - output_path (str): The path to the file where the content should be written.
                          If None, the content is printed to the console.
 
@@ -22,3 +24,9 @@ def write_output(content, output_path):
             click.echo(f"Error writing to output file: {e}", err=True)
     else:
         click.echo(content)
+
+    # Copy content to clipboard
+    try:
+        pyperclip.copy(content)
+    except Exception as e:
+        click.echo(f"Error copying to clipboard: {e}", err=True)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code2prompt"
-version = "0.6.0"
+version = "0.6.1"
 description = ""
 authors = ["Raphael MANSUY <raphael.mansuy@gmail.com>"]
 readme = "README.md"
@@ -12,6 +12,7 @@ click = "^8.1.7"
 jinja2 = "^3.1.4"
 prompt-toolkit = "^3.0.47"
 tiktoken = "^0.7.0"
+pyperclip = "^1.9.0"
 
 [tool.poetry.scripts]
 code2prompt = "code2prompt.main:create_markdown_file"