File tree Expand file tree Collapse file tree 2 files changed +45
-0
lines changed Expand file tree Collapse file tree 2 files changed +45
-0
lines changed Original file line number Diff line number Diff line change 1
1
# Specify files that shouldn't be modified by Fern
2
+ src/pipedream/pipedream.py
Original file line number Diff line number Diff line change
1
+ import os
2
+ from string import Template
3
+
4
+ from .client import (
5
+ AsyncClient ,
6
+ Client ,
7
+ )
8
+ from .environment import PipedreamEnvironment
9
+
10
+
11
+ class Pipedream (Client ):
12
+ def __init__ (
13
+ self ,
14
+ project_id : str ,
15
+ environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
16
+ * args ,
17
+ ** kwargs ,
18
+ ):
19
+ super ().__init__ (base_url = _get_base_url (environment ), * args , ** kwargs )
20
+ self .project_id = project_id
21
+
22
+
23
+ class AsyncPipedream (AsyncClient ):
24
+ def __init__ (
25
+ self ,
26
+ project_id : str ,
27
+ environment : PipedreamEnvironment = PipedreamEnvironment .PROD ,
28
+ * args ,
29
+ ** kwargs ,
30
+ ):
31
+ super ().__init__ (base_url = _get_base_url (environment ), * args , ** kwargs )
32
+ self .project_id = project_id
33
+
34
+
35
+ def _get_base_url (environment : PipedreamEnvironment ) -> str :
36
+ if not environment :
37
+ raise Exception ("Please pass environment to construct the client" )
38
+
39
+ user = os .getenv ("DEV_NAMESPACE" , "" )
40
+ return Template (environment .value ).substitute (
41
+ {
42
+ "user" : user ,
43
+ }
44
+ )
You can’t perform that action at this time.
0 commit comments