Add ability to set api token via environment variable (#13)

This commit is contained in:
Jared Zhao
2022-04-27 12:19:15 -07:00
committed by GitHub
parent 293370d5af
commit 14c4704fb9
4 changed files with 18 additions and 2 deletions

View File

@@ -15,7 +15,9 @@ function Dashboard(props: Props) {
const [data, setData] = useState<{[name: string]: Data}>();
const getData = () => {
getDataAPI(Date.now() / 1000 - (60 * 60)).then((data) => {
// TODO: Retrieve all data for now, until we implement selector
// e.g. Something like: Date.now() / 1000 - (60 * 60)
getDataAPI(0).then((data) => {
setData(data.data);
});
}

View File

@@ -6,3 +6,4 @@ SQL_MODE = "sqlite"
SQL_VERSION = 0
DEFAULT_USER = True
DEFAULT_API_KEY = os.environ.get("DEFAULT_API_KEY", "default")

View File

@@ -14,6 +14,15 @@ def create_user(email: str) -> uuid.UUID:
return user_id
def get_user_id(email: str) -> uuid.UUID:
"""
Get a user.
"""
user_id = queries.get_user_id(email=email)
return user_id
def generate_api_token(user_id: uuid.UUID, api_token: Optional[str] = None) -> None:
"""
Generate a api_token for a user.

View File

@@ -127,10 +127,14 @@ def startup() -> None:
if config.DEFAULT_USER:
try:
user_id = auth.create_user(email="default")
auth.generate_api_token(user_id=user_id, api_token="default")
except:
# Default user already exists
pass
finally:
# Update default token (in case it changed)
user_id = auth.get_user_id(email="default")
auth.generate_api_token(user_id=user_id, api_token=config.DEFAULT_API_KEY)
logger.info("")
logger.info("READY")