def generate_meta(gen, pattern):
    """Generate metadata from name and insert into the clip"""

    if not gen.status(True):
        return (False, "Generation not running.")


    if not os.path.isdir(epp_root):
        return (False, "Could not find EPP ROOT directory. Set the EPP_ROOT env variable first.", )

    PAT_FILEBASENAME = re.compile(pattern)
    selected_versions = gen.selected_versions()

    if len(selected_versions) < 1:
        return (False, "No versions selected.")

    for version in selected_versions:
        meta = version.Metadata(gen.frame())
        path = meta['Data']['File']['Path']

        mat = PAT_FILEBASENAME.findall(path)
        if len(mat):
            shotname = mat[0]
            gen.add_meta(version, shotname)
            log.info("Setting metadata of {0} to {1}".format(version.Name, shotname))
        else:
            log.warning("Skipping " + os.path.basename(path))

    return (True, "")
Example #2
0
def generate_meta(gen, pattern):
    """Generate metadata from name and insert into the clip"""

    if not gen.status(True):
        return (False, "Generation not running.")

    if not os.path.isdir(epp_root):
        return (
            False,
            "Could not find EPP ROOT directory. Set the EPP_ROOT env variable first.",
        )

    PAT_FILEBASENAME = re.compile(pattern)
    selected_versions = gen.selected_versions()

    if len(selected_versions) < 1:
        return (False, "No versions selected.")

    for version in selected_versions:
        meta = version.Metadata(gen.frame())
        path = meta['Data']['File']['Path']

        mat = PAT_FILEBASENAME.findall(path)
        if len(mat):
            shotname = mat[0]
            gen.add_meta(version, shotname)
            log.info("Setting metadata of {0} to {1}".format(
                version.Name, shotname))
        else:
            log.warning("Skipping " + os.path.basename(path))

    return (True, "")
Example #3
0
    def _setup(self):
        """
        Setup the widgets
        """

        epp_root = osplus.get_env("EPP_ROOT")

        if not os.path.isdir(epp_root):
            QMessageBox.critical(
                self, "Error",
                "Could not find EPP ROOT directory.\n{0}".format(epp_root))
            return False

        self.settings = settings.XMLSettings(
            os.path.join(epp_root, "config.xml"))
        self.PROJECTROOT = self.settings.get("paths", "projectdir")

        if self.PROJECTROOT is None:
            QMessageBox.critical(
                self, "Error",
                "No Project Directory configured in config.xml.")
            return False

        self._setup_header(epp_root)
        self._setup_status()

        # Require Generation
        if not self._setup_generation():
            msg = "Could not establish connection with Generation. Make sure it is running."
            log.error(msg, True)
            return False

        # Need edit rights for this one!
        if not self._gen.allow_changes(True):
            log.warning("Project locked.")
            return False

        if not self._setup_project():
            msg = "Project directory from Generation Project invalid: {0}".format(
                self.PROJECTDIR)
            log.error(msg, True)
            return False

        if not self._setup_shot():
            msg = "Project directory from Generation Project invald: {0}".format(
                self.PROJECTDIR)
            log.error(msg, True)
            return False

        return True
Example #4
0
    def _setup(self):
        """
        Setup the widgets
        """

        epp_root = osplus.get_env("EPP_ROOT")

        if not os.path.isdir(epp_root):
            QMessageBox.critical(self, "Error", "Could not find EPP ROOT directory.\n{0}".format(epp_root))
            return False

        self.settings = settings.XMLSettings(os.path.join(epp_root, "config.xml") )
        self.PROJECTROOT = self.settings.get("paths", "projectdir")

        if self.PROJECTROOT is None:
            QMessageBox.critical(self, "Error", "No Project Directory configured in config.xml.")
            return False

        self._setup_header(epp_root)
        self._setup_status()


        # Require Generation
        if not self._setup_generation():
            msg = "Could not establish connection with Generation. Make sure it is running."
            log.error(msg, True)
            return False

        # Need edit rights for this one!
        if not self._gen.allow_changes(True):
            log.warning("Project locked.")
            return False

        if not self._setup_project():
            msg = "Project directory from Generation Project invalid: {0}".format(self.PROJECTDIR)
            log.error(msg, True)
            return False


        if not self._setup_shot():
            msg = "Project directory from Generation Project invald: {0}".format(self.PROJECTDIR)
            log.error(msg, True)
            return False

        return True
Example #5
0
    def insert_media(self, shotname, filepath, only_matching=False):
        """docstring for insert_comp"""
        clipm = self.get_shot_clip(shotname)
        proj = self.app.ProjectGet()
        sub = proj.SubGet()
        track = sub.TrackGet()

        is_comp = os.path.splitext(filepath)[1].lower() == ".comp"

        if clipm is None:
            log.warning(
                "Shot {0} not found in Generation project.".format(shotname))
            return False

        mat = self.PAT_VER.match(os.path.basename(filepath))
        if mat is not None:
            target_base, target_ver, target_post = mat.groups()

        index = 0

        for j in range(int(clipm.VersionCount())):
            cur_ver = clipm.VersionGet(j)
            loader = cur_ver.GetLoader()
            cur_filepath = loader.Filename
            if loader is None:
                continue

            if is_comp:
                cur_filepath = loader.RefFilename()

            if cur_filepath is None:
                continue

            if cur_filepath.lower() == filepath.lower():
                log.debug("Media {0} already exists".format(
                    os.path.basename(filepath)))
                return False

            mat = self.PAT_VER.match(os.path.basename(cur_filepath))
            if mat is not None and target_ver is not None:
                base, ver, post = mat.groups()
                if int(ver) == int(target_ver) - 1:
                    index = cur_ver.Version

        if only_matching:
            if not os.path.basename(filepath).lower().startswith(
                    shotname.lower().replace(os.path.sep, "_")):
                log.debug("Skipping because not a shot media {0}".format(
                    os.path.basename(filepath)))
                return False

        #lock = self.app.UpdateLock()

        for i, loader in proj.DropFiles(str(filepath)).items():
            log.debug("Inserting {0} at {1}".format(os.path.basename(filepath),
                                                    index))
            # Latest one.
            last_ver = clipm.VersionGet(0)
            in_point = 0
            if last_ver is not None:
                in_point = last_ver.InPoint

            new_ver = clipm.VersionInsert(loader)
            #TODO: new_ver = clipm.VersionInsert(loader, index)
            new_ver.InPoint = in_point

        #self.app.UpdateUnlock(lock)

        return True
Example #6
0
    def _setup(self):
        """
        Setup the widgets
        """

        epp_root = osplus.get_env("EPP_ROOT")

        if not os.path.isdir(epp_root):
            log.error("Could not find EPP ROOT directory.\n{0}".format(epp_root), True)
            return False

        self.settings = settings.XMLSettings(os.path.join(epp_root, "config.xml") )
        self.PROJECTROOT = self.settings.get("paths", "projectdir")
        if not self.PROJECTROOT.endswith(os.path.sep):
            self.PROJECTROOT = self.PROJECTROOT + os.path.sep 

        if self.PROJECTROOT is None:
            log.error("No Project Directory configured in config.xml.", True)
            return False


        self._setup_header(epp_root)
        #self._setup_name_widget(epp_root)
        self._setup_status()

        # Require Generation
        if not self._setup_generation():
            msg = "Could not establish connection with Generation. Make sure it is running."
            log.error(msg, True)
            return False

        # Need edit rights for this one!
        if not self._gen.allow_changes(True):
            return False

        if not self._setup_project():
            msg = "Project directory from Generation Project invalid: {0}".format(self.PROJECTDIR)
            log.error(msg, True)
            return False

        if not self._setup_shot():
            msg = "Project directory from Generation Project invald: {0}".format(self.PROJECTDIR)
            log.error(msg, True)
            return False

        selected_versions = self._gen.selected_versions()
        if len(selected_versions) < 1:
            msg = "No versions selected."
            log.error(msg, True)
            return False

        meta = selected_versions[0].Metadata(selected_versions[0].InPoint)
        self.preview_path = meta['Data']['File']['Path']

        if not self._setup_patterns(epp_root):
            msg = "No parsing patterns found."
            log.warning(msg)
            return False

        # We can't work without structs
        if not self._setup_structs(epp_root):
            msg = "Could not find directory structure files in templates.\n{0}".format(os.path.join(epp_root, "templates", "shot_dirs"))
            log.error(msg, True)
            return False

        return True
Example #7
0
    def insert_media(self, shotname, filepath, only_matching=False):
        """docstring for insert_comp"""
        clipm = self.get_shot_clip(shotname)
        proj = self.app.ProjectGet()
        sub = proj.SubGet()
        track = sub.TrackGet()

        is_comp = os.path.splitext(filepath)[1].lower() == ".comp"

        if clipm is None:
            log.warning("Shot {0} not found in Generation project.".format(shotname))
            return False

        mat = self.PAT_VER.match(os.path.basename(filepath))
        if mat is not None:
            target_base, target_ver, target_post = mat.groups()

        index = 0

        for j in range(int(clipm.VersionCount())):
            cur_ver = clipm.VersionGet(j)
            loader = cur_ver.GetLoader()
            cur_filepath = loader.Filename
            if loader is None:
                continue

            if is_comp:
                cur_filepath = loader.RefFilename()

            if cur_filepath is None:
                continue

            if cur_filepath.lower() == filepath.lower():
                log.debug("Media {0} already exists".format(os.path.basename(filepath)))
                return False


            mat = self.PAT_VER.match(os.path.basename(cur_filepath))
            if mat is not None and target_ver is not None:
                base, ver, post = mat.groups()
                if int(ver) == int(target_ver)-1:
                    index = cur_ver.Version

        if only_matching:
            if not os.path.basename(filepath).lower().startswith(shotname.lower().replace(os.path.sep, "_")):
                log.debug("Skipping because not a shot media {0}".format(os.path.basename(filepath)))
                return False

        #lock = self.app.UpdateLock()

        for i, loader in proj.DropFiles(str(filepath)).items():
            log.debug("Inserting {0} at {1}".format(os.path.basename(filepath), index))
            # Latest one.
            last_ver = clipm.VersionGet(0)
            in_point = 0
            if last_ver is not None:
                in_point = last_ver.InPoint

            new_ver = clipm.VersionInsert(loader)
            #TODO: new_ver = clipm.VersionInsert(loader, index)
            new_ver.InPoint = in_point

        #self.app.UpdateUnlock(lock)
    
        return True
Example #8
0
    def _setup(self):
        """
        Setup the widgets
        """

        epp_root = osplus.get_env("EPP_ROOT")

        if not os.path.isdir(epp_root):
            log.error(
                "Could not find EPP ROOT directory.\n{0}".format(epp_root),
                True)
            return False

        self.settings = settings.XMLSettings(
            os.path.join(epp_root, "config.xml"))
        self.PROJECTROOT = self.settings.get("paths", "projectdir")
        if not self.PROJECTROOT.endswith(os.path.sep):
            self.PROJECTROOT = self.PROJECTROOT + os.path.sep

        if self.PROJECTROOT is None:
            log.error("No Project Directory configured in config.xml.", True)
            return False

        self._setup_header(epp_root)
        #self._setup_name_widget(epp_root)
        self._setup_status()

        # Require Generation
        if not self._setup_generation():
            msg = "Could not establish connection with Generation. Make sure it is running."
            log.error(msg, True)
            return False

        # Need edit rights for this one!
        if not self._gen.allow_changes(True):
            return False

        if not self._setup_project():
            msg = "Project directory from Generation Project invalid: {0}".format(
                self.PROJECTDIR)
            log.error(msg, True)
            return False

        if not self._setup_shot():
            msg = "Project directory from Generation Project invald: {0}".format(
                self.PROJECTDIR)
            log.error(msg, True)
            return False

        selected_versions = self._gen.selected_versions()
        if len(selected_versions) < 1:
            msg = "No versions selected."
            log.error(msg, True)
            return False

        meta = selected_versions[0].Metadata(selected_versions[0].InPoint)
        self.preview_path = meta['Data']['File']['Path']

        if not self._setup_patterns(epp_root):
            msg = "No parsing patterns found."
            log.warning(msg)
            return False

        # We can't work without structs
        if not self._setup_structs(epp_root):
            msg = "Could not find directory structure files in templates.\n{0}".format(
                os.path.join(epp_root, "templates", "shot_dirs"))
            log.error(msg, True)
            return False

        return True