main.py

"""
URL Shortener
Shortens long URLs using various services
"""

import pyshorteners


def shorten_url(long_url, service='tinyurl'):
    """Shorten URL"""
    try:
        shortener = pyshorteners.Shortener()
        
        if service == 'tinyurl':
            short_url = shortener.tinyurl.short(long_url)
        elif service == 'isgd':
            short_url = shortener.isgd.short(long_url)
        else:
            short_url = shortener.tinyurl.short(long_url)
        
        return short_url
    except Exception as e:
        return f"Error: {e}"


def main():
    """Main function"""
    print("\n" + "="*60)
    print("           URL SHORTENER")
    print("="*60 + "\n")
    
    try:
        while True:
            url = input("Enter URL to shorten (or 'quit'): ").strip()
            
            if url.lower() in ['quit', 'exit', 'q']:
                print("\nGoodbye!")
                break
            
            if not url:
                continue
            
            if not url.startswith(('http://', 'https://')):
                url = 'https://' + url
            
            print("\nShortening URL...")
            short = shorten_url(url)
            
            print("\n" + "="*60)
            print(f"Original: {url}")
            print(f"Shortened: {short}")
            print("="*60 + "\n")
    
    except KeyboardInterrupt:
        print("\n\nGoodbye!")


if __name__ == "__main__":
    main()