#!/usr/bin/env python

import sys
import hashlib

usage = """
Uso:
	hash [mode] [string]
	hash [mode]

Mode:
        -m   calculate md5
        -s   calculate sha1
	-a   calculate both
"""

try:
    	sys.argv[1]
except:
	print (usage)
	mode = input("Select a mode: ")
	if mode == "-s":
		passwd = input("Insert a string to being hashed with sha1sum: ")
		print (hashlib.sha1(passwd.encode("utf-8")).hexdigest(), "sha1")
		exit()
	elif mode == "-a":
		passwd = input("Insert a string to being hashed with both functions: ")
		print (hashlib.md5(passwd.encode("utf-8")).hexdigest(), "md5")
		print (hashlib.sha1(passwd.encode("utf-8")).hexdigest(), "sha1")
		exit()
	else:
		passwd = input("Insert a string to being hashed with md5: ")
		print (hashlib.md5(passwd.encode("utf-8")).hexdigest(), "md5")
		exit()
try:
	sys.argv[2]
	if sys.argv[1] == "-s":
		print (hashlib.sha1(sys.argv[2].encode("utf-8")).hexdigest(), "sha1")
		exit()
	elif sys.argv[1] == "-a":
		print (hashlib.md5(sys.argv[2].encode("utf-8")).hexdigest(), "md5")
		print (hashlib.sha1(sys.argv[2].encode("utf-8")).hexdogest(), "sha1")
		exit()
	else:
		print (hashlib.md5(sys.argv[2].encode("utf-8")).hexdigest(), "md5")
		exit()
except:
	if sys.argv[1] == "-s":
		passwd = input("Insert a string to being hashed with sha1sum: ")
		print (hashlib.sha1(passwd.encode("utf-8")).hexdigest(), "sha1")
		exit()
	elif sys.argv[1] == "-a":
		passwd = input("Insert a string to being hashed with both functions: ")
		print (hashlib.md5(passwd.encode("utf-8")).hexdigest(), "md5")
		print (hashlib.sha1(passwd.encode("utf-8")).hexdigest(), "sha1")
		exit()
	else:
		passwd = input("Insert a string to being hashed with md5: ")
		print (hashlib.md5(passwd.encode("utf-8")).hexdigest(), "md5")
		exit()