lucky_reactor / app.py
elfsong
Update
55f36d8
raw
history blame
1.12 kB
import os
import streamlit as st
import googleapiclient.discovery
from google.oauth2 import service_account
def auth():
with open("/tmp/token.json", "w") as token_f:
google_key = os.getenv("GOOGLE_KEY")
token_f.write(google_key)
credentials = service_account.Credentials.from_service_account_file('/tmp/token.json')
return credentials
def get_status(credentials):
status = False
client = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
result = client.instances().list(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-c').execute()
for item in result['items']:
if item["name"] == "oj":
status = True if item["status"] == "RUNNING" else False
return status
def activate_server():
return True
st.title("Lucky Reactor")
credentials = auth()
status = True
st.markdown("credentials: " + credentials)
st.image("cover.jpg", caption="Reactor is currently " + "running πŸš€" if status else "sleeping 😴" + ".")
if not status:
if st.button("Ignite"):
st.write("Here you go: [LINK]")