Close session on exit

Author: averyanalex

README.md

@@ -38,6 +38,7 @@ from shazamio import Shazam
 async def main():
     shazam = Shazam()
     out = await shazam.recognize_song('dora.ogg')
+    await shazam.stop()
     print(out)
 
 loop = asyncio.get_event_loop()
@@ -63,6 +64,7 @@ async def main():
     artist_id = 43328183
     about_artist = await shazam.artist_about(artist_id)
     serialized = Serialize.artist(about_artist)
+    await shazam.stop()
 
     print(about_artist)  # dict
     print(serialized)  # serialized from dataclass factory
@@ -91,6 +93,7 @@ async def main():
     track_id = 552406075
     about_track = await shazam.track_about(track_id=track_id)
     serialized = Serialize.track(data=about_track)
+    await shazam.stop()
 
     print(about_track)  # dict
     print(serialized)  # serialized from dataclass factory
@@ -119,6 +122,7 @@ async def main():
     shazam = Shazam()
     track_id = 559284007
     count = await shazam.listening_counter(track_id=track_id)
+    await shazam.stop()
     print(count)
 
 loop = asyncio.get_event_loop()
@@ -144,6 +148,7 @@ async def main():
     track_id = 546891609
     related = await shazam.related_tracks(track_id=track_id, limit=5, offset=2)
     # ONLY №3, №4 SONG
+    await shazam.stop()
     print(related)
 
 loop = asyncio.get_event_loop()
@@ -168,6 +173,7 @@ async def main():
     for artist in artists['artists']['hits']:
         serialized = Serialize.artist(data=artist)
         print(serialized)
+    await shazam.stop()
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())
@@ -190,6 +196,7 @@ from shazamio import Shazam
 async def main():
     shazam = Shazam()
     tracks = await shazam.search_track(query='Lil', limit=5)
+    await shazam.stop()
     print(tracks)
 
 loop = asyncio.get_event_loop()
@@ -229,6 +236,8 @@ async def main():
     for i in serialized.data[0].views.top_songs.data:
         print(i.attributes.name)
 
+    await shazam.stop()
+
 
 loop = asyncio.get_event_loop_policy().get_event_loop()
 loop.run_until_complete(main())
@@ -260,6 +269,8 @@ async def main():
         # SERIALIZE FROM DATACLASS FACTORY
         print(serialized)
 
+    await shazam.stop()
+
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())
 
@@ -285,6 +296,7 @@ async def main():
     for track in top_five_track_from_amsterdam['tracks']:
         serialized = Serialize.track(data=track)
         print(serialized)
+    await shazam.stop()
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())
@@ -311,6 +323,7 @@ async def main():
         genre=GenreMusic.HIP_HOP_RAP,
         limit=4
     )
+    await shazam.stop()
     print(top_spain_rap)
 
 loop = asyncio.get_event_loop()
@@ -339,6 +352,8 @@ async def main():
         serialized_track = Serialize.track(data=track)
         print(serialized_track.spotify_url)
 
+    await shazam.stop()
+
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())
@@ -365,6 +380,7 @@ async def main():
     for track in top_world_tracks['tracks']:
         serialized = Serialize.track(track)
         print(serialized)
+    await shazam.stop()
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())
@@ -390,6 +406,7 @@ async def main():
     for track in top_five_track_from_amsterdam['tracks']:
         serialized = Serialize.track(data=track)
         print(serialized.title)
+    await shazam.stop()
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(main())

shazamio/api.py

@@ -27,6 +27,9 @@ def __init__(self, language: str = "en-US", endpoint_country: str = "GB"):
         self.language = language
         self.endpoint_country = endpoint_country
 
+    async def stop(self):
+        super().stop()
+
     async def top_world_tracks(self, limit: int = 200, offset: int = 0) -> Dict[str, Any]:
         """
         Search top world tracks

shazamio/client.py

@@ -5,11 +5,12 @@
 
 
 class HTTPClient:
-    http_session: aiohttp.ClientSession
-
     def __init__(self, *args, **kwargs):
         self.http_session = aiohttp.ClientSession()
 
+    async def stop(self):
+        await self.http_session.close()
+
     async def request(self, method: str, url: str, *args, **kwargs) -> dict:
         if method.upper() == "GET":
             async with self.http_session.get(url, **kwargs) as resp:

shazamio/converter.py

@@ -8,6 +8,9 @@
 
 
 class Geo(HTTPClient):
+    async def stop(self):
+        super().stop()
+
     async def city_id_from(self, country: Union[CountryCode, str], city: str) -> int:
         """
         Return City ID from country name and city name.

tests/test_recognize.py

@@ -16,6 +16,7 @@ async def song_bytes():
 async def test_recognize_song_file():
     shazam = Shazam()
     out = await shazam.recognize_song(data="examples/data/dora.ogg")
+    await shazam.stop()
 
     assert out.get("matches") != []
     assert out["track"]["key"] == "549679333"
@@ -25,6 +26,7 @@ async def test_recognize_song_file():
 async def test_recognize_song_bytes(song_bytes: bytes):
     shazam = Shazam()
     out = await shazam.recognize_song(data=song_bytes)
+    await shazam.stop()
 
     assert out.get("matches") != []
     assert out["track"]["key"] == "549679333"
@@ -42,6 +44,7 @@ async def test_recognize_song_too_short():
 
     shazam = Shazam()
     out = await shazam.recognize_song(data=short_audio_segment)
+    await shazam.stop()
 
     assert out.get("matches") == []
     assert "track" not in out