elfsong commited on
Commit
786017a
ยท
1 Parent(s): 5c3da8a
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -12,26 +12,27 @@ def auth():
12
 
13
  return credentials
14
 
15
- def get_status(credentials):
16
  status = False
17
  client = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
18
 
19
  result = client.instances().list(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-b').execute()
20
  for item in result['items']:
21
- if item["name"] == "venus":
22
  status = True if item["status"] == "RUNNING" else False
23
  return status
24
 
25
- def activate_server():
26
  service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
27
- request = service.instances().resume(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-b', instance='venus')
28
  response = request.execute()
29
  return response
30
 
31
  st.title("Lucky Reactor")
32
 
 
33
  credentials = auth()
34
- status = get_status(credentials)
35
 
36
  st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running ๐Ÿš€" if status else "sleeping ๐Ÿ˜ด") + ".")
37
 
@@ -42,5 +43,5 @@ if not status:
42
 
43
  if submitted and token == os.getenv("EASY_TOKEN"):
44
  st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")
45
- response = activate_server()
46
- st.write(response)
 
12
 
13
  return credentials
14
 
15
+ def get_status(credentials, instance):
16
  status = False
17
  client = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
18
 
19
  result = client.instances().list(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-b').execute()
20
  for item in result['items']:
21
+ if item["name"] == instance:
22
  status = True if item["status"] == "RUNNING" else False
23
  return status
24
 
25
+ def activate_server(credentials, instance):
26
  service = googleapiclient.discovery.build('compute', 'v1', credentials=credentials)
27
+ request = service.instances().resume(project='nus-cisco-corp-lab-wp1', zone='asia-southeast1-b', instance=instance)
28
  response = request.execute()
29
  return response
30
 
31
  st.title("Lucky Reactor")
32
 
33
+ instance = "lucky-reactor"
34
  credentials = auth()
35
+ status = get_status(credentials, instance)
36
 
37
  st.image("cover.jpg", caption="Lucky Reactor is currently " + ("running ๐Ÿš€" if status else "sleeping ๐Ÿ˜ด") + ".")
38
 
 
43
 
44
  if submitted and token == os.getenv("EASY_TOKEN"):
45
  st.write("Lucky Reactor has been ignited. Please wait a few minutes (3~5) to preceed...")
46
+ response = activate_server(credentials, instance)
47
+ # st.write(response)