mirror of
https://github.com/jetfuel-dev/jetfuel.git
synced 2025-12-16 16:04:51 -06:00
Add ability to set api token via environment variable (#13)
This commit is contained in:
@@ -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);
|
||||
});
|
||||
}
|
||||
|
||||
@@ -6,3 +6,4 @@ SQL_MODE = "sqlite"
|
||||
SQL_VERSION = 0
|
||||
|
||||
DEFAULT_USER = True
|
||||
DEFAULT_API_KEY = os.environ.get("DEFAULT_API_KEY", "default")
|
||||
|
||||
@@ -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.
|
||||
|
||||
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user