only return IP if either IP or hostname is supplied

return_only_ip.py

#!/usr/bin/python3
# @atucom
''' parse simple command line arguments '''
import argparse
parser = argparse.ArgumentParser(description='only return IP if either IP or hostname is supplied')
parser.add_argument('target', help='IP or hostname')
args = parser.parse_args()


def ip_filter(input):
    ''' return ip if already an ip, otherwise resolve hostname to ip '''
    import socket
    try:
        ip = socket.gethostbyname(input)
        return ip
    except socket.gaierror as e:
        print("BAD INPUT: {}".format(input))
    except:
        return input

ip = ip_filter(args.target) 
print(ip)