class UserCache:
def __init__(self):
self.name_provider = NameProvider()
self.cached = RedisSet(self.name_provider.set_names.cached_ips)
def is_cached(self, ip_address):
"""
Returns whether or not IP address is cached
:param ip_address: IP address to check
:return: boolean indicating if IP is cached
"""
key = self._get_key_for(ip_address)
return self.cached.exists(key)
def get(self, ip_address):
"""
Returns cached data for IP address
:param ip_address: IP address to get cached data for
:return: dict containing cached data
"""
key = self._get_key_for(ip_address)
return RedisHash(key).to_dict()
def set(self, ip_address, data):
"""
Sets cached data for given IP address
:param ip_address: IP address to set cached data for
:param data: data to set in cache
"""
key = self._get_key_for(ip_address)
return RedisHash(key).update(data)
def delete(self, ip_address, key_to_delete):
"""
Removes (key, value) pair from IP address hash
:param ip_address: IP address of cached data
:param key_to_delete: key to delete
"""
ip_address_key = self._get_key_for(ip_address)
return RedisHash(ip_address_key).delete(key_to_delete)
def _get_key_for(self, ip_address):
"""
Returns redis key for ip address cached data
"""
return "{}{}".format(self.name_provider.prefixes.ip, ip_address)
class UsernameGenerator:
def __init__(self):
self.displayNames = RedisSet("displayNames")
self.nouns = []
self.adjectives = []
with open("resources/nouns.txt", "r") as f:
self.nouns = [line.strip('\n') for line in f if line.strip('\n')]
with open("resources/adjectives_names.txt", "r") as f:
self.adjectives = [line.strip('\n') for line in f if line]
def get_username(self):
return random.choice(self.adjectives) + random.choice(self.nouns)
def get_usernames_in_bulk(self, count=50):
usernames = []
while len(usernames) < count:
username = self.get_username()
if not self.displayNames.exists(username):
usernames += [username]
return usernames