Add -r / --re-encrypt

This commit is contained in:
Nichlas Severinsen 2021-06-02 20:25:49 +02:00
parent 46317877c1
commit 5e9bd4f257
6 changed files with 93 additions and 12 deletions

View file

@ -43,7 +43,15 @@ if __name__ == '__main__':
optional.add_argument('-d', '--decryption-key', dest='decryption_key', type=str, help='Manually specify key', default='')
optional.add_argument('-v', '--verbose', dest='verbose', help='Increase verbosity', action='count')
optional.add_argument('-q', '--quiet', dest='quiet', help='Quiet mode, only prints on error', action='store_true')
# -e is reserved for "extract" so re-encrypt is "-r"
optional.add_argument('-r', '--re-encrypt', dest='reencrypt', help='Re-encrypt .iso', action='store_true')
args = parser.parse_args()
core.decrypt(args)
if args.reencrypt:
core.encrypt(args)
else:
core.decrypt(args)