moira-python-client 4.2.1__py3-none-any.whl → 4.3.1__py3-none-any.whl

moira_client/models/contact.py

@@ -23,6 +23,7 @@ class Contact(Base):
         self.user = kwargs.get('user', None)
         self.name = kwargs.get('name', None)
         self._id = kwargs.get('id', None)
+        self.team_id = kwargs.get('team_id', None)
 
 
 class ContactManager:

moira_client/models/subscription.py

@@ -2,12 +2,10 @@ from ..client import InvalidJSONError
 from ..client import ResponseStructureError
 from .base import Base
 
-class Subscription(Base):
-    def __init__(self, client, tags, contacts=None, enabled=None, throttling=None, sched=None,
+class SubscriptionModel(Base):
+    def __init__(self, tags, contacts=None, enabled=None, throttling=None, sched=None,
                  ignore_warnings=False, ignore_recoverings=False, plotting=None, any_tags=False, **kwargs):
         """
-
-        :param client: api client
         :param tags: list of str tags
         :param contacts: list of contact id's
         :param enabled: bool is enabled
@@ -19,8 +17,6 @@ class Subscription(Base):
         :param any_tags: bool any tags
         :param kwargs: additional parameters
         """
-        self._client = client
-
         self._id = kwargs.get('id', None)
         self.tags = tags if not any_tags else []
         if not contacts:
@@ -38,32 +34,7 @@ class Subscription(Base):
         if not plotting:
             plotting = {'enabled': False, 'theme': 'light'}
         self.plotting = plotting
-
-    def _send_request(self, subscription_id=None):
-        data = {
-            'contacts': self.contacts,
-            'tags': self.tags,
-            'enabled': self.enabled,
-            'any_tags': self.any_tags,
-            'throttling': self.throttling,
-            'sched': self.sched,
-            'ignore_warnings': self.ignore_warnings,
-            'ignore_recoverings': self.ignore_recoverings,
-            'plotting': self.plotting
-        }
-
-        if subscription_id:
-            data['id'] = subscription_id
-
-        if subscription_id:
-            result = self._client.put('subscription/{id}'.format(id=subscription_id), json=data)
-        else:
-            result = self._client.put('subscription', json=data)
-        if 'id' not in result:
-            raise ResponseStructureError("id doesn't exist in response", result)
-
-        self._id = result['id']
-        return self._id
+        self.team_id = kwargs.get('team_id', None)
 
     def disable_day(self, day):
         """
@@ -124,26 +95,6 @@ class Subscription(Base):
             'theme': 'light'
             }
 
-    def save(self):
-        """
-        Save subscription
-
-        :return: subscription id
-        """
-        if self._id:
-            return self.update()
-        self._send_request()
-
-    def update(self):
-        """
-        Update subscription
-
-        :return: subscription id
-        """
-        if not self._id:
-            return self.save()
-        self._send_request(self._id)
-
     def set_start_hour(self, hour):
         """
         Set start hour
@@ -185,6 +136,85 @@ class Subscription(Base):
         self._end_minute = minute
 
 
+class Subscription(SubscriptionModel):
+    def __init__(self, client, tags, contacts=None, enabled=None, throttling=None, sched=None,
+                 ignore_warnings=False, ignore_recoverings=False, plotting=None, any_tags=False, **kwargs):
+        """
+        :param client: api client
+        :param tags: list of str tags
+        :param contacts: list of contact id's
+        :param enabled: bool is enabled
+        :param throttling: bool throttling
+        :param sched: dict schedule
+        :param ignore_warnings: bool ignore warnings
+        :param ignore_recoverings: bool ignore recoverings
+        :param plotting: dict plotting settings
+        :param any_tags: bool any tags
+        :param kwargs: additional parameters
+        """
+        self._client = client
+
+        super().__init__(
+            tags=tags,
+            contacts=contacts,
+            enabled=enabled,
+            throttling=throttling,
+            sched=sched,
+            ignore_warnings=ignore_warnings,
+            ignore_recoverings=ignore_recoverings,
+            plotting=plotting,
+            any_tags=any_tags,
+            **kwargs,
+        )
+
+    def _send_request(self, subscription_id=None):
+        data = {
+            'contacts': self.contacts,
+            'tags': self.tags,
+            'enabled': self.enabled,
+            'any_tags': self.any_tags,
+            'throttling': self.throttling,
+            'sched': self.sched,
+            'ignore_warnings': self.ignore_warnings,
+            'ignore_recoverings': self.ignore_recoverings,
+            'plotting': self.plotting,
+            'team_id': self.team_id,
+        }
+
+        if subscription_id:
+            data['id'] = subscription_id
+
+        if subscription_id:
+            result = self._client.put('subscription/{id}'.format(id=subscription_id), json=data)
+        else:
+            result = self._client.put('subscription', json=data)
+        if 'id' not in result:
+            raise ResponseStructureError("id doesn't exist in response", result)
+
+        self._id = result['id']
+        return self._id
+
+    def save(self):
+        """
+        Save subscription
+
+        :return: subscription id
+        """
+        if self._id:
+            return self.update()
+        self._send_request()
+
+    def update(self):
+        """
+        Update subscription
+
+        :return: subscription id
+        """
+        if not self._id:
+            return self.save()
+        self._send_request(self._id)
+
+
 class SubscriptionManager:
     def __init__(self, client):
         self._client = client

moira_client/models/team/__init__.py

@@ -0,0 +1,8 @@
+from ._managers import TeamManager
+from ._models import TeamModel
+
+
+__all__ = [
+    "TeamManager",
+    "TeamModel",
+]

moira_client/models/team/_managers.py

@@ -0,0 +1,96 @@
+from typing import Optional
+
+from ._models import TeamModel, UserTeams, SaveTeamResponse
+from .contact import TeamContactManager
+from .settings import TeamSettingsManager
+from .subscription import TeamSubscriptionManager
+from .user import TeamUserManager
+from ...client import Client
+
+
+class TeamManager:
+    def __init__(self, client: Client) -> None:
+        self._client = client
+        self._user = None  # type: Optional[TeamUserManager]
+        self._settings = None  # type: Optional[TeamSettingsManager]
+        self._subscription = None  # type: Optional[TeamSubscriptionManager]
+        self._contact = None  # type: Optional[TeamContactManager]
+
+    def get_all(self) -> UserTeams:
+        """Get all teams"""
+        response = self._client.get(self._full_path())
+
+        return UserTeams(teams=[TeamModel(**team) for team in response["teams"]])
+
+    def create(self, team: TeamModel) -> SaveTeamResponse:
+        """Create a new team"""
+        payload = {
+            "name": team.name,
+            "description": team.description,
+        }
+
+        response = self._client.post(self._full_path(), json=payload)
+
+        return SaveTeamResponse(**response)
+
+    def delete(self, team_id: str) -> SaveTeamResponse:
+        """Delete a team"""
+        response = self._client.delete(self._full_path(team_id))
+
+        return SaveTeamResponse(**response)
+
+    def get(self, team_id: str) -> TeamModel:
+        """Get a team by ID"""
+        response = self._client.get(self._full_path(team_id))
+
+        return TeamModel(**response)
+
+    def update(self, team_id: str, team: TeamModel) -> SaveTeamResponse:
+        """Update existing team"""
+        payload = {
+            "name": team.name,
+            "description": team.description,
+        }
+
+        response = self._client.put(self._full_path(team_id), json=payload)
+
+        return SaveTeamResponse(**response)
+
+    @property
+    def user(self) -> TeamUserManager:
+        """Get team user manager"""
+        if self._user is None:
+            self._user = TeamUserManager(self._client)
+
+        return self._user
+
+    @property
+    def settings(self) -> TeamSettingsManager:
+        """Get team settings manager"""
+        if self._settings is None:
+            self._settings = TeamSettingsManager(self._client)
+
+        return self._settings
+
+    @property
+    def subscription(self) -> TeamSubscriptionManager:
+        """Get team subscription manager"""
+        if self._subscription is None:
+            self._subscription = TeamSubscriptionManager(self._client)
+
+        return self._subscription
+
+    @property
+    def contact(self) -> TeamContactManager:
+        """Get team contact manager"""
+        if self._contact is None:
+            self._contact = TeamContactManager(self._client)
+
+        return self._contact
+
+    @staticmethod
+    def _full_path(team_id: Optional[str] = None) -> str:
+        if team_id is None:
+            return "teams"
+
+        return "teams/{team_id}".format(team_id=team_id)

moira_client/models/team/_models.py

@@ -0,0 +1,19 @@
+from ..base import Base
+from typing import List
+
+
+class TeamModel(Base):
+    def __init__(self, description: str, name: str, **kwargs):
+        self.description = description
+        self.name = name
+        self._id = kwargs.get("id", None)
+
+
+class UserTeams:
+    def __init__(self, teams: List[TeamModel]):
+        self.teams = teams
+
+
+class SaveTeamResponse(Base):
+    def __init__(self, **kwargs):
+        self._id = kwargs.get("id")

moira_client/models/team/contact/__init__.py

@@ -0,0 +1,5 @@
+from ._managers import TeamContactManager
+
+__all__ = [
+    "TeamContactManager",
+]

moira_client/models/team/contact/_managers.py

@@ -0,0 +1,24 @@
+from ...contact import Contact
+from ....client import Client
+
+
+class TeamContactManager:
+    def __init__(self, client: Client) -> None:
+        self._client = client
+
+    def create(self, team_id: str, contact: Contact) -> Contact:
+        """Create a new team contact"""
+        payload = {
+            "team_id": team_id,
+            "name": contact.name,
+            "type": contact.type,
+            "value": contact.value,
+        }
+
+        response = self._client.post(self._full_path(team_id), json=payload)
+
+        return Contact(**response)
+
+    @staticmethod
+    def _full_path(team_id: str) -> str:
+        return "teams/{team_id}/contacts".format(team_id=team_id)

moira_client/models/team/settings/__init__.py

@@ -0,0 +1,7 @@
+from ._managers import TeamSettingsManager
+from ._models import TeamSettings
+
+__all__ = [
+    "TeamSettings",
+    "TeamSettingsManager",
+]

moira_client/models/team/settings/_managers.py

@@ -0,0 +1,23 @@
+from ._models import TeamSettings
+from ...contact import Contact
+from ...subscription import Subscription
+from ....client import Client
+
+
+class TeamSettingsManager:
+    def __init__(self, client: Client) -> None:
+        self._client = client
+
+    def get(self, team_id: str) -> TeamSettings:
+        """Get team settings"""
+        response = self._client.get(self._full_path(team_id))
+
+        return TeamSettings(
+            contacts=[Contact(**contact) for contact in response["contacts"]],
+            subscriptions=[Subscription(self._client, **subscription) for subscription in response["subscriptions"]],
+            team_id=response["team_id"],
+        )
+
+    @staticmethod
+    def _full_path(team_id: str) -> str:
+        return "teams/{team_id}/settings".format(team_id=team_id)

moira_client/models/team/settings/_models.py

@@ -0,0 +1,11 @@
+from typing import List
+
+from ...contact import Contact
+from ...subscription import Subscription
+
+
+class TeamSettings:
+    def __init__(self, contacts: List[Contact], subscriptions: List[Subscription], team_id: str) -> None:
+        self.contacts = contacts
+        self.subscriptions = subscriptions
+        self.team_id = team_id

moira_client/models/team/subscription/__init__.py

@@ -0,0 +1,5 @@
+from ._managers import TeamSubscriptionManager
+
+__all__ = [
+    "TeamSubscriptionManager",
+]

moira_client/models/team/subscription/_managers.py

@@ -0,0 +1,30 @@
+from ...subscription import SubscriptionModel, Subscription
+from ....client import Client
+
+
+class TeamSubscriptionManager:
+    def __init__(self, client: Client) -> None:
+        self._client = client
+
+    def create(self, team_id: str, subscription: SubscriptionModel) -> Subscription:
+        """Create a new team subscription"""
+        payload = {
+            "team_id": team_id,
+            "contacts": subscription.contacts,
+            "tags": subscription.tags,
+            "enabled": subscription.enabled,
+            "any_tags": subscription.any_tags,
+            "throttling": subscription.throttling,
+            "sched": subscription.sched,
+            "ignore_warnings": subscription.ignore_warnings,
+            "ignore_recoverings": subscription.ignore_recoverings,
+            "plotting": subscription.plotting,
+        }
+
+        response = self._client.post(self._full_path(team_id), json=payload)
+
+        return Subscription(self._client, **response)
+
+    @staticmethod
+    def _full_path(team_id: str) -> str:
+        return "teams/{team_id}/subscriptions".format(team_id=team_id)

moira_client/models/team/user/__init__.py

@@ -0,0 +1,7 @@
+from ._managers import TeamUserManager
+from ._models import TeamMembers
+
+__all__ = [
+    "TeamMembers",
+    "TeamUserManager",
+]

moira_client/models/team/user/_managers.py

@@ -0,0 +1,43 @@
+from ._models import TeamMembers
+from ....client import Client
+from typing import Optional
+
+
+class TeamUserManager:
+    def __init__(self, client: Client) -> None:
+        self._client = client
+
+    def get(self, team_id: str) -> TeamMembers:
+        """Get users of a team"""
+        response = self._client.get(self._full_path(team_id))
+
+        return TeamMembers(**response)
+
+    def add(self, team_id: str, members: TeamMembers) -> TeamMembers:
+        """Add users to a team"""
+        payload = {"usernames": members.usernames}
+
+        response = self._client.post(self._full_path(team_id), json=payload)
+
+        return TeamMembers(**response)
+
+    def set(self, team_id: str, members: TeamMembers) -> TeamMembers:
+        """Set users of a team"""
+        payload = {"usernames": members.usernames}
+
+        response = self._client.put(self._full_path(team_id), json=payload)
+
+        return TeamMembers(**response)
+
+    def delete(self, team_id: str, team_user_id: str) -> TeamMembers:
+        """Delete a user from a team"""
+        response = self._client.delete(self._full_path(team_id, team_user_id))
+
+        return TeamMembers(**response)
+
+    @staticmethod
+    def _full_path(team_id: str, team_user_id: Optional[str] = None) -> str:
+        if team_user_id is None:
+            return "teams/{team_id}/users".format(team_id=team_id)
+
+        return "teams/{team_id}/users/{team_user_id}".format(team_id=team_id, team_user_id=team_user_id)

moira_client/models/team/user/_models.py

@@ -0,0 +1,6 @@
+from typing import List
+
+
+class TeamMembers:
+    def __init__(self, usernames: List[str]) -> None:
+        self.usernames = usernames

moira_client/moira.py

@@ -9,6 +9,7 @@ from .models.trigger import TriggerManager
 from .models.health import HealthManager
 from .models.config import ConfigManager
 from .models.user import UserManager
+from .models.team import TeamManager
 
 
 class Moira:
@@ -34,6 +35,7 @@ class Moira:
         self._health = None
         self._config = None
         self._user = None
+        self._team = None
 
     @property
     def trigger(self):
@@ -145,3 +147,12 @@ class Moira:
         if not self._user:
             self._user = UserManager(self._client)
         return self._user
+
+    @property
+    def team(self) -> TeamManager:
+        """Get team manager
+        """
+        if not self._team:
+            self._team = TeamManager(self._client)
+
+        return self._team

{moira_python_client-4.2.1.dist-info → moira_python_client-4.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: moira-python-client
-Version: 4.2.1
+Version: 4.3.1
 Summary: Client for Moira - Alerting system based on Graphite data
 Home-page: https://github.com/moira-alert/python-moira-client
 Author: Alexander Lukyanchenko

moira_python_client-4.3.1.dist-info/RECORD

@@ -0,0 +1,33 @@
+moira_client/__init__.py,sha256=BUWUwGU1V5wvdi5-mUXd8AJlsGyrAvFKOjs8-vkUV4o,67
+moira_client/client.py,sha256=SGakaO2SB2EjsZPlfD1Dv2naIgwBr5F5QufYKm-Ro1w,4106
+moira_client/moira.py,sha256=Pa3grEVhGu2gcPxWR8sdu7GGowNF3AXBYvK1nmRiZi8,3859
+moira_client/models/__init__.py,sha256=K3Kcjh8F_jV703ii-QMeN9cMwqcCfbfeNF7e9MfCD1c,126
+moira_client/models/base.py,sha256=cYVLxnc5PGjmUcoZRZ7AFx5f0Ij96HypJpOjpwr_D-Q,385
+moira_client/models/config.py,sha256=BbsShuKhe5H0DTXkIeOndHyXbi2xW2BRP0pwoIQl4Ok,2178
+moira_client/models/contact.py,sha256=-z8p32EIUfG9x8KRbUPeJR3DroOuPHfBN9pv8yq4lgI,4987
+moira_client/models/event.py,sha256=pXsuU-X3i3DX89rs1wBaB7SBvw_q5dgJwLJgQFrp7vg,1375
+moira_client/models/health.py,sha256=a2bGgHVVbcIJCraArjI9V-gqpwdRZ1B5SnNrsLqR6mI,1708
+moira_client/models/notification.py,sha256=stk78RW0lENkBX7FHm9cTskHkz8SxdXmpcXP72XwWio,1946
+moira_client/models/pattern.py,sha256=Ef8Roupkjrs8gb41vrgBwvgpgTmJU4igsU58z19LHcg,1619
+moira_client/models/subscription.py,sha256=vAGuxrcTkGLtrj8D_cC_lo3EMGtAnClOXdmFfMl45E0,9050
+moira_client/models/tag.py,sha256=2sDmf9CKAzzZRqQykBdR3-jKgcfo62hGO7eivI1X_sc,4336
+moira_client/models/trigger.py,sha256=61Gs4blN0pcPrCEnOzpaKkA_UgeChz3Kv2qV-zXySYE,17868
+moira_client/models/user.py,sha256=9zG7pbmEA9ESUWFMzvqfyLP0lqcjz1pc0n8U7mL1UMk,1767
+moira_client/models/team/__init__.py,sha256=hPeciH0IcuSdo3jY7EMyYaG85faZky4VvU9BsGDpZiY,118
+moira_client/models/team/_managers.py,sha256=plbYKrocnD7ae4E8K7_RtPeRjn2JXQcQaKVULw5jwIA,2991
+moira_client/models/team/_models.py,sha256=swwR1g7ZlxoV_wbmbEW_ADqnnaZXJLb5fyH8utptEx0,438
+moira_client/models/team/contact/__init__.py,sha256=JyUCAhv_KmTjzPz1Ej38uJdzcDvFFguemc1ZjpRs6fk,83
+moira_client/models/team/contact/_managers.py,sha256=zOEL4vXnUsD76I3I4aYa9dyeqIVvlIGKN2LBu8MhOaU,679
+moira_client/models/team/settings/__init__.py,sha256=IX7rZVpaKY45uB11lRwDcgumnbJnlJy2cEQkLF6qm-k,139
+moira_client/models/team/settings/_managers.py,sha256=12vnS6D8LicVzW5rMV74OdIQipEqPD9Vd96BRYKukxA,791
+moira_client/models/team/settings/_models.py,sha256=MWdCSiacf1p2LTsTN1yrtZ4Ga8452nhRwIUMLceRqNM,332
+moira_client/models/team/subscription/__init__.py,sha256=DIQSml5LNW7NvB5MjEGK5sEsfos3jIxiul_rO8VKU-Q,93
+moira_client/models/team/subscription/_managers.py,sha256=lMWyWc8jibQcjLOBAXTKtCY2mH5Gs0F8cB8fvSHU4Gg,1103
+moira_client/models/team/user/__init__.py,sha256=XDgaELIXAcD8l0SBVjQ8OSaSknMd1qGJ_dRmT0VxyVI,129
+moira_client/models/team/user/_managers.py,sha256=9AeMrMVjek7hsHS9mGNeX9RaK6UCV5JU6GlFDouI_b0,1458
+moira_client/models/team/user/_models.py,sha256=E90MLjV4Xtbypy_LXatzJdmCy1MUs8NZfs0RlRneUsI,134
+moira_python_client-4.3.1.dist-info/LICENSE,sha256=HJk93uYDrJpwAa6G3ufYfC4ZgCKH2flOtdgOv1esxgI,1062
+moira_python_client-4.3.1.dist-info/METADATA,sha256=k58zTJCXroosf0-24sD2c6ezDI-q-dgTbuuUe_UXEJ0,1243
+moira_python_client-4.3.1.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+moira_python_client-4.3.1.dist-info/top_level.txt,sha256=zBB6_FP28feugBNn2cLmi5nCniRPnuChI6HuXUqFdIA,13
+moira_python_client-4.3.1.dist-info/RECORD,,

{moira_python_client-4.2.1.dist-info → moira_python_client-4.3.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

moira_python_client-4.2.1.dist-info/RECORD

@@ -1,20 +0,0 @@
-moira_client/__init__.py,sha256=BUWUwGU1V5wvdi5-mUXd8AJlsGyrAvFKOjs8-vkUV4o,67
-moira_client/client.py,sha256=SGakaO2SB2EjsZPlfD1Dv2naIgwBr5F5QufYKm-Ro1w,4106
-moira_client/moira.py,sha256=40OmxdnUTyrhjZhDwhy9OToeZhYyMLkytB619X0p8z4,3601
-moira_client/models/__init__.py,sha256=K3Kcjh8F_jV703ii-QMeN9cMwqcCfbfeNF7e9MfCD1c,126
-moira_client/models/base.py,sha256=cYVLxnc5PGjmUcoZRZ7AFx5f0Ij96HypJpOjpwr_D-Q,385
-moira_client/models/config.py,sha256=BbsShuKhe5H0DTXkIeOndHyXbi2xW2BRP0pwoIQl4Ok,2178
-moira_client/models/contact.py,sha256=iBt9A-wOF6CVprRfUVIKeJMkZy4vxb7zuvFxdE23x0E,4936
-moira_client/models/event.py,sha256=pXsuU-X3i3DX89rs1wBaB7SBvw_q5dgJwLJgQFrp7vg,1375
-moira_client/models/health.py,sha256=a2bGgHVVbcIJCraArjI9V-gqpwdRZ1B5SnNrsLqR6mI,1708
-moira_client/models/notification.py,sha256=stk78RW0lENkBX7FHm9cTskHkz8SxdXmpcXP72XwWio,1946
-moira_client/models/pattern.py,sha256=Ef8Roupkjrs8gb41vrgBwvgpgTmJU4igsU58z19LHcg,1619
-moira_client/models/subscription.py,sha256=xyB8MXM6mC0VjE_2aezBnSSaFER9u4LAQ0kJuzJsxas,7891
-moira_client/models/tag.py,sha256=2sDmf9CKAzzZRqQykBdR3-jKgcfo62hGO7eivI1X_sc,4336
-moira_client/models/trigger.py,sha256=61Gs4blN0pcPrCEnOzpaKkA_UgeChz3Kv2qV-zXySYE,17868
-moira_client/models/user.py,sha256=9zG7pbmEA9ESUWFMzvqfyLP0lqcjz1pc0n8U7mL1UMk,1767
-moira_python_client-4.2.1.dist-info/LICENSE,sha256=HJk93uYDrJpwAa6G3ufYfC4ZgCKH2flOtdgOv1esxgI,1062
-moira_python_client-4.2.1.dist-info/METADATA,sha256=dZBENvHfgEfmaYvqZCDRl1wgvV4yBqRlONoHD32HNwA,1243
-moira_python_client-4.2.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-moira_python_client-4.2.1.dist-info/top_level.txt,sha256=zBB6_FP28feugBNn2cLmi5nCniRPnuChI6HuXUqFdIA,13
-moira_python_client-4.2.1.dist-info/RECORD,,