Change behavior, add collector when not assigned on announce

pylibagent/agent.py

@@ -96,14 +96,35 @@ async def announce(self, asset_name: Optional[str] = None,
             async with ClientSession(headers=self._headers) as session:
                 async with session.get(
                         url,
-                        params={'field': 'name'},
+                        params={'fields': 'name', 'collectors': 'key'},
                         ssl=self.verify_ssl) as r:
                     if r.status != 200:
                         msg = await r.text()
                         raise Exception(f'{msg} (error code: {r.status})')
 
                     resp = await r.json()
                     name = resp["name"]
+                    collectors = resp["collectors"]
+
+            for collector in collectors:
+                if collector['key'] == self.key:
+                    break
+            else:
+                # The collector is not assigned yet
+                url = _join(
+                    self.api_uri,
+                    f'asset/{self.asset_id}/collector/{self.key}')
+                try:
+                    async with ClientSession(headers=self._headers) as session:
+                        async with session.post(url, ssl=self.verify_ssl) as r:
+                            if r.status != 204:
+                                msg = await r.text()
+                                raise Exception(
+                                    f'{msg} (error code: {r.status})')
+                except Exception as e:
+                    msg = str(e) or type(e).__name__
+                    logging.error(f'failed to assign collector: {msg}')
+
             logging.info(f'announced agent {name} (Id: {self.asset_id})')
             return
 

pylibagent/version.py

@@ -1 +1 @@
-__version__ = '0.1.9'
+__version__ = '0.2.0'