Initial commit

pdfuck.py

@@ -0,0 +1,38 @@
+# pdfuck.py
+# Author JiJi <i@mmdjiji.com>
+# GitHub https://github.com/mmdjiji/pdfuck
+
+import argparse
+import pikepdf
+
+def pdfuck(filepath, output, password):
+
+  # add '.fuck' sign for the output file
+  if not output:
+    splited = filepath.rsplit('.', 1)
+    prefix = splited[0]
+    output = prefix + '.fucked'
+    if(len(splited) >= 2):
+      suffix = splited[1]
+      output += '.' + suffix
+
+  try:
+    with pikepdf.open(filename_or_stream=filepath, password=password) as pdf:
+      pdf.save(filename_or_stream=output)
+    print('Decryption succeeded, saved to \'' + output + '\'')
+
+  except Exception as e:
+    print(e)
+
+def main():
+  parser = argparse.ArgumentParser(description='PDFuck: Remove the password of your PDF file')
+  parser.add_argument('filepath', metavar='<filepath>', type=str, help='path of the source PDF file')
+  parser.add_argument('-o', '--output', metavar='<filepath>', type=str, help='path of the target PDF file')
+  parser.add_argument('-p', '--password', metavar='<password>', type=str, help='reading password of the source PDF file')
+  args = parser.parse_args()
+
+  filepath, output, password = args.filepath, args.output, args.password or ''
+  pdfuck(filepath=filepath, output=output, password=password)
+
+if __name__ == '__main__':
+  main()