33import aenum
44from Cryptodome .PublicKey import RSA
55
6- from bunq .sdk import model
6+ from bunq .sdk . model import core
77from bunq .sdk import security
88from bunq .sdk .json import converter
9- from bunq .sdk .model import generated
9+ from bunq .sdk .model . generated import endpoint
1010
1111
1212class ApiEnvironmentType (aenum .AutoNumberEnum ):
@@ -95,7 +95,7 @@ def _initialize_installation(self):
9595 """
9696
9797 private_key_client = security .generate_rsa_private_key ()
98- installation = model .Installation .create (
98+ installation = core .Installation .create (
9999 self ,
100100 security .public_key_to_string (private_key_client .publickey ())
101101 ).value
@@ -119,12 +119,12 @@ def _register_device(self, device_description,
119119 :rtype: None
120120 """
121121
122- generated .DeviceServer .create (
122+ endpoint .DeviceServer .create (
123123 self ,
124124 {
125- generated .DeviceServer .FIELD_DESCRIPTION : device_description ,
126- generated .DeviceServer .FIELD_SECRET : self .api_key ,
127- generated .DeviceServer .FIELD_PERMITTED_IPS : permitted_ips ,
125+ endpoint .DeviceServer .FIELD_DESCRIPTION : device_description ,
126+ endpoint .DeviceServer .FIELD_SECRET : self .api_key ,
127+ endpoint .DeviceServer .FIELD_PERMITTED_IPS : permitted_ips ,
128128 }
129129 )
130130
@@ -133,7 +133,7 @@ def _initialize_session(self):
133133 :rtype: None
134134 """
135135
136- session_server = model .SessionServer .create (self ).value
136+ session_server = core .SessionServer .create (self ).value
137137 token = session_server .token .token
138138 expiry_time = self ._get_expiry_timestamp (session_server )
139139
@@ -142,7 +142,7 @@ def _initialize_session(self):
142142 @classmethod
143143 def _get_expiry_timestamp (cls , session_server ):
144144 """
145- :type session_server: model .SessionServer
145+ :type session_server: core .SessionServer
146146
147147 :rtype: datetime.datetime
148148 """
@@ -155,7 +155,7 @@ def _get_expiry_timestamp(cls, session_server):
155155 @classmethod
156156 def _get_session_timeout_seconds (cls , session_server ):
157157 """
158- :type session_server: model .SessionServer
158+ :type session_server: core .SessionServer
159159
160160 :rtype: int
161161 """
@@ -210,7 +210,7 @@ def _delete_session(self):
210210 :rtype: None
211211 """
212212
213- generated .Session .delete (self , self ._SESSION_ID_DUMMY )
213+ endpoint .Session .delete (self , self ._SESSION_ID_DUMMY )
214214
215215 @property
216216 def environment_type (self ):
@@ -300,7 +300,7 @@ def restore(cls, path=None):
300300
301301 with open (path , cls ._FILE_MODE_READ ) as file_ :
302302 return cls .from_json (file_ .read ())
303-
303+
304304 @classmethod
305305 def from_json (cls , json_str ):
306306 """
0 commit comments