From 3fb4cce881683906955eea8ff8bd07225d8ca8b7 Mon Sep 17 00:00:00 2001
From: gd <gechandev@gmail.com>
Date: Thu, 20 May 2021 10:47:28 +0300
Subject: [PATCH] init

---
 pwc.py | 85 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100755 pwc.py

diff --git a/pwc.py b/pwc.py
new file mode 100755
index 0000000..2d5ffde
--- /dev/null
+++ b/pwc.py
@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+
+import sys
+import getopt
+import subprocess
+
+import random
+import string
+
+def generate_pass(special_chars: bool, lenght: int, print_pass: bool) -> str:
+    abc = string.ascii_letters + string.digits
+    if special_chars:
+        abc = abc + string.punctuation
+    pw = ''
+    for char in range(lenght):
+        pw = pw + random.choice(abc)
+
+    if print_pass:
+        print(pw)
+    else:
+        # Run xclip. Copy generated password to X clipboard!
+        command = 'xclip -selection clipboard'.split()
+        process = subprocess.Popen(
+            command,
+            stdin=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE
+        )
+        process.stdin.write(pw.encode())  # Put password to STDIN.
+
+def usage():
+    print('Usage: pwc [-h | --help] [-s | --special] [-p | --print] [<lenght>]')
+
+def print_help():
+    print('Generate password and copy it to clipboard.')
+    print()
+    usage()
+    print()
+    print('    -s, --special    use punctuation characters in password.')
+    print('    -p, --print      print password to STDOUT instead of copy to clipboard.')
+    print('    -h, --help       print this help message and exit.')
+
+def main():
+    # Set default values.
+    special_chars = False
+    print_pass = False
+    lenght = 12
+
+    # Parse agrs.
+    try:
+        opts, args = getopt.getopt(
+            sys.argv[1:], 
+            "hsp", 
+            ["help", "special", "print"]
+        )
+    except getopt.GetoptError as err:
+        print(err)  # will print something like "option -a not recognized"
+        usage()
+        sys.exit(2)
+        special_chars=False
+    for opt, arg in opts:
+        if opt in ("-h", "--help"):
+            print_help()
+            sys.exit()
+        if opt in ("-s", "--special"):
+            special_chars=True
+        if opt in ("-p", "--print"):
+            print_pass=True
+    if len(args) >= 2:
+        print('Too many arguments.')
+        usage()
+        sys.exit(2)
+    for arg in args:
+        try:
+            lenght = int(arg)
+        except ValueError:
+            print('Argument must be an integer.')
+            usage()
+            sys.exit(1)
+
+    # Run password generator.
+    generate_pass(special_chars, lenght, print_pass)
+
+if __name__ == "__main__":
+    main()