535. Encode and Decode TinyURL

class Codec:
    code_db, url_db = {}, {}
    chars = string.ascii_letters + string.digits

    def getCode(self) -> str:
        code = ''.join(random.choice(self.chars) for i in range(6))
        return code
 
    def encode(self, longUrl: str) -> str:
        """Encodes a URL to a shortened URL.
        """
        if longUrl in self.url_db: 
            return self.url_db[longUrl]
        code = self.getCode()
        self.code_db[code] = longUrl
        self.url_db[longUrl] = code
        return code

    def decode(self, shortUrl: str) -> str:
        """Decodes a shortened URL to its original URL.
        """
        return self.code_db[shortUrl]
# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.decode(codec.encode(url))