import os
import uuid
import smartninja_redis
redis = smartninja_redis.from_url(os.environ.get("REDIS_URL"))
def create_csrf_token(username):
csrf_token = str(uuid.uuid4())
redis.set(name=csrf_token, value=username)
return csrf_token
def validate_csrf(csrf, username):
redis_csrf_username = redis.get(name=csrf) # csrf from Redis
# if CSRF token is found in cache
if redis_csrf_username:
csrf_username = redis_csrf_username.decode(
) # needs to be decoded from byte string
return username == csrf_username
else:
return False
def set_last_bitt(bitt_id):
redis.set(name="last-bitt-id", value=str(bitt_id))
def get_last_bitt():
import smartninja_redis as redis
from smartninja_redis import Redis
# from_url()
r_url = redis.from_url(url="localhost")
r_url.set(name="Ninja", value="Smart")
print(r_url.get("Ninja"))
# Redis class
r_class = Redis(host="localhost")
r_class.set(name="smart", value="ninja")
print(r_class.get("smart"))
r_class.set(name="smart", value="samurai")
print(r_class.get("smart"))