delphix_snapshot_group_will_plugin.py

#!/usr/bin/env python
# For use with HipChat and Will
# https://github.com/skoczen/will

from will.decorators import hear
from will.decorators import periodic
from will.decorators import randomly
from will.decorators import rendered_template
from will.decorators import require_settings
from will.decorators import respond_to
from will.decorators import route
from will.plugin import WillPlugin

from delphixpy.v1_6_0 import job_context
from delphixpy.v1_6_0.delphix_engine import DelphixEngine
from delphixpy.v1_6_0.web import database
from delphixpy.v1_6_0.web import group


class DelphixSnapshotPlugin(WillPlugin):
    @respond_to("snapshot_group (?P<v_object>.*)")
    def snapshot_group_will(self, message, v_object=None):
        group_name = v_object
        # database_name = "Employee DB - Dev"

        server_session = DelphixEngine(
            "landsharkengine", "delphix_admin", "landshark", "DOMAIN"
        )

        all_groups = group.get_all(server_session)

        for each in all_groups:
            if group_name == each.name:
                group_reference = each.reference
                break

        database_objs = database.get_all(server_session, group=group_reference)

        with job_context.asyncly(server_session):
            for obj in database_objs:
                database.sync(server_session, obj.reference)