Merge pull request #42 from 0ladyshek/master

Добавлена авторизация по ID школы

netschoolapi/netschoolapi.py

@@ -37,7 +37,7 @@ async def __aenter__(self) -> 'NetSchoolAPI':
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         await self.logout()
 
-    async def login(self, user_name: str, password: str, school: str):
+    async def login(self, user_name: str, password: str, school: Union[str, int]):
         response_with_cookies = await self._client.get('logindata')
         self._client.cookies.extract_cookies(response_with_cookies)
 
@@ -228,14 +228,14 @@ async def logout(self):
         await self._client.post('auth/logout')
         await self._client.aclose()
 
-    async def _address(self, school: str) -> Dict[str, int]:
+    async def _address(self, school: Union[str, int]) -> Dict[str, int]:
         response = await self._client.get(
             'addresses/schools', params={'funcType': 2}
         )
 
         schools_reference = response.json()
         for school_ in schools_reference:
-            if school_['name'] == school:
+            if school_['name'] == school or school_['id'] == school:
                 self._school_id = school_['id']
                 return {
                     'cid': school_['countryId'],