11import sys
22import requests
3+ from constants import VERSION
34
45
56class TorqueSession (requests .Session ):
67 def __init__ (self ):
78 super (TorqueSession , self ).__init__ ()
8- self .headers .update ({"Accept" : "application/json" , "Accept-Charset" : "utf-8" })
9+ self .headers .update (
10+ {
11+ "Accept" : "application/json" ,
12+ "Accept-Charset" : "utf-8" ,
13+ "User-Agent" : f"Torque-Plugin-Github-Actions/{ VERSION } "
14+ }
15+ )
916
1017 def torque_auth (self , token : str ) -> None :
1118 self .headers .update ({"Authorization" : f"Bearer { token } " })
1219
1320
1421class TorqueClient :
15- def __init__ (self , space : str , token : str , session : TorqueSession = TorqueSession (), account : str = None ):
22+ TORQUE_SERVER = "portal.qtorque.io"
23+
24+ def __init__ (self , space : str , token : str , session : TorqueSession = TorqueSession ()):
1625 self .token = token
1726 self .space = space
1827 self .session = session
1928 session .torque_auth (self .token )
20- self .base_api_url = f"https://qtorque.io /api/spaces/{ self .space } "
29+ self .base_api_url = f"https://{ self . TORQUE_SERVER } /api/spaces/{ self .space } "
2130
2231 def _request (self , endpoint : str , method : str = 'GET' , params : dict = None ) -> requests .Response :
2332 self ._validate_creds ()
@@ -56,23 +65,22 @@ def start_sandbox(
5665 sandbox_name : str ,
5766 duration : int = 120 ,
5867 inputs : dict = None ,
59- artifacts : dict = None ,
60- branch : str = None ) -> str :
68+ # branch: str = None
69+ ) -> str :
6170
62- path = "sandbox "
71+ path = "environments "
6372 iso_duration = f"PT{ duration } M"
6473 params = {
6574 "sandbox_name" : sandbox_name ,
6675 "blueprint_name" : blueprint_name ,
6776 "duration" : iso_duration ,
6877 "inputs" : inputs ,
69- "artifacts" : artifacts ,
7078 }
7179
72- if branch :
73- params ["source" ] = {
74- "branch" : branch ,
75- }
80+ # if branch:
81+ # params["source"] = {
82+ # "branch": branch,
83+ # }
7684
7785 res = self ._request (path , method = "POST" , params = params )
7886 sandbox_id = res .json ()["id" ]
@@ -81,15 +89,15 @@ def start_sandbox(
8189
8290 def get_sandbox (self , sandbox_id : str ) -> dict :
8391 """Returns Sandbox as a json"""
84- path = f"sandbox /{ sandbox_id } "
92+ path = f"environments /{ sandbox_id } "
8593
8694 res = self ._request (path , method = "GET" )
8795
8896 return res .json ()
8997
9098
9199 def end_sandbox (self , sandbox_id : str ) -> None :
92- path = f"sandbox /{ sandbox_id } "
100+ path = f"environments /{ sandbox_id } "
93101
94102 res = self ._request (path , method = "DELETE" )
95103
0 commit comments