import datetime


def set_cookie(response, key, value, day_expire=1):
    max_age = day_expire * 24 * 60 * 60
    expires = datetime.datetime.strftime(datetime.datetime.utcnow() + datetime.timedelta(seconds=max_age),
                                         "%a, %d-%b-%Y %H:%M:%S GMT")
    response.set_cookie(key=key, value=value, max_age=max_age, expires=expires)  # @TODO: Connect cookie to domain


def get_cookie(cookies, key):
    return cookies.get(key) if cookies.get(key) else None


def pop_cookie(response, key):
    cookie = get_cookie(response.cookies, key)
    if cookie:
        response.delete_cookie(key)
    return cookie