File tree Expand file tree Collapse file tree 2 files changed +46
-0
lines changed Expand file tree Collapse file tree 2 files changed +46
-0
lines changed Original file line number Diff line number Diff line change 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