import os

from maya import cmds

import avalon.maya
import pype.api


class ExtractRig(pype.api.Extractor):
    """Extract rig as Maya Ascii"""

    label = "Extract Rig (Maya ASCII)"
    hosts = ["maya"]
    families = ["rig"]

    def process(self, instance):

        # Define extract output file path
        dir_path = self.staging_dir(instance)
        filename = "{0}.ma".format(instance.name)
        path = os.path.join(dir_path, filename)

        # Perform extraction
        self.log.info("Performing extraction..")
        with avalon.maya.maintained_selection():
            cmds.select(instance, noExpand=True)
            cmds.file(path,
                      force=True,
                      typ="mayaAscii",
                      exportSelected=True,
                      preserveReferences=False,
                      channels=True,
                      constraints=True,
                      expressions=True,
                      constructionHistory=True)

        if "representations" not in instance.data:
            instance.data["representations"] = []

        representation = {
            'name': 'ma',
            'ext': 'ma',
            'files': filename,
            "stagingDir": dir_path
        }
        instance.data["representations"].append(representation)


        self.log.info("Extracted instance '%s' to: %s" % (instance.name, path))