Initial commit

utilities/fcpx.py

@@ -0,0 +1,200 @@
+import xml.etree.ElementTree as ET
+import argparse
+import os
+import glob
+from urllib2 import unquote
+import time
+
+library = None
+
+
+class FCPXLibrary:
+
+    def __init__(self, xml_path):
+        parser = ET.parse(xml_path)
+        self.root = parser.getroot()
+        self.xml_version = self.root.attrib.get('version')
+        self.location = self.library_location()
+
+        # self.projects = self.root.findall('./library/event/project')
+        self.formats = self.root.findall('./resources/format')
+
+        self.clips = [Clip(x, self) for x in self.root.findall(".//asset-clip")]
+        self.projects = [Project(x, self) for x in self.root.findall('./library/event/project')]
+
+    def formats(self):
+        return self.root.findall('./resources/format')
+
+    def element_with_tag_value(self, element, tag, value):
+        return self.root.findall(".//{e}[@{t}='{v}']".format(e=element, t=tag, v=value))
+
+    def clips_with_videorole(self, role):
+        return [clip for clip in self.clips if getattr(clip, 'videoRole', None) == role]
+
+    def format_with_id(self, id):
+        # return self.root.findall("./resources/format[id='{}']".format(id))
+        return self.element_with_tag_value('format', 'id', id)
+
+    def library_location(self):
+        # urllib2.unquote(asset_ref.get('src'))[7:]
+        path = self.root.findall('./library')[0].attrib['location']
+        return unquote(path)[7:]
+
+
+class Project:
+    def __init__(self, project_element, library):
+        for attrib in project_element.attrib:
+            setattr(self, attrib, project_element.get(attrib))
+        print(project_element.attrib)
+        print(project_element.parent)
+
+        ref_clips = project_element.findall(".//ref-clip")
+
+        print('start')
+        for clip in library.clips:
+            print(clip.name)
+            if clip.name == ref_clips[0]:
+                print(clip)
+                break
+        print('end')
+
+        # for child in ref_clips:
+        #     print(child.tag, child.attrib)
+
+
+class Clip:
+    def __init__(self, clip_element, library):
+
+        # self.library = library
+
+        # Get attribs from XML
+        for attrib in clip_element.attrib:
+            setattr(self, attrib, clip_element.get(attrib))
+        self.type = 'audio' if hasattr(self, 'audioRole') else 'video'
+        
+
+        # Get clip reference
+        asset_ref = next(iter(library.element_with_tag_value('asset', 'id', self.ref)))
+
+        for attrib in asset_ref.attrib:
+            if not hasattr(self, attrib):
+                setattr(self, attrib, asset_ref.get(attrib))
+
+        self.source = unquote(asset_ref.get('src'))[7:]
+
+        if self.type == 'video':
+
+            format_id = getattr(self, 'format', asset_ref.get('format', None))
+            video_format = next(iter(library.format_with_id(format_id)))
+
+            if not hasattr(self, 'format'):
+                print('no format!')
+
+            try:
+                frame_duration = fcp_time_to_float(video_format.get('frameDuration'))
+                self.in_frame = int(round(fcp_time_to_float(self.start) / frame_duration))
+                duration = int(round(fcp_time_to_float(self.duration) / frame_duration))
+                self.out_frame = self.in_frame + duration
+            except Exception as e:
+                print('in/out fail: ' + str(e))
+                print(dir(self))
+                pass
+
+    def optimized_source(self):
+        path = None
+        mov = os.path.splitext(os.path.basename(self.source))[0] + '.mov'
+        found = glob.glob(os.path.join(library.location, '*', 'Transcoded Media', 'High Quality Media', mov))
+        if found:
+            path = found[0]
+            print(path)
+        return path
+
+    def proxy_source(self):
+        path = None
+        mov = os.path.splitext(os.path.basename(self.source))[0] + '.mov'
+        found = glob.glob(os.path.join(library.location, '*', 'Transcoded Media', 'Proxy Media', mov))
+        if found:
+            path = found[0]
+            print(path)
+        return path
+
+    def __repr__(self):
+        if self.type == 'video':
+            return "<Clip name:'%s' type: %s role: '%s' duration:%s frames>" % (getattr(self, 'name', None), self.type,
+                                                                              getattr(self, 'videoRole', None), self.out_frame - self.in_frame)
+        else:
+            return "<Clip name:'%s' type: %s role: '%s'>" % (getattr(self, 'name', None), self.type, getattr(self, 'audioRole', None))
+
+
+def fcp_time_to_float(timestr):
+    try:
+        rates = timestr.strip('s').split('/')
+        return float(rates[0]) / float(rates[-1])
+    except (ZeroDivisionError, AttributeError) as e:
+        return 0.0
+
+import sys
+from types import ModuleType, FunctionType
+from gc import get_referents
+
+# Custom objects know their class.
+# Function objects seem to know way too much, including modules.
+# Exclude modules as well.
+BLACKLIST = type, ModuleType, FunctionType
+
+
+def getsize(obj):
+    """sum size of object & members."""
+    if isinstance(obj, BLACKLIST):
+        raise TypeError('getsize() does not take argument of type: '+ str(type(obj)))
+    seen_ids = set()
+    size = 0
+    objects = [obj]
+    while objects:
+        need_referents = []
+        for obj in objects:
+            if not isinstance(obj, BLACKLIST) and id(obj) not in seen_ids:
+                seen_ids.add(id(obj))
+                size += sys.getsizeof(obj)
+                need_referents.append(obj)
+        objects = get_referents(*need_referents)
+    return size
+
+
+# if __name__ == "__main__":
+#
+#     parser = argparse.ArgumentParser()
+#     parser.add_argument('-i', '--input', help='Input FCPX Library XML', required=True)
+#     parser.add_argument('-s', '--save-file', help='Description', required=False)
+#
+#     args = parser.parse_args()
+#
+#     library = FCPXLibrary(args.input)
+#
+#     print getsize(library)
+#     while True:
+#         time.sleep(4)
+#
+#     print library.library_location()
+#
+#     print dir(library.clips[0])
+#     print library.clips[0]
+#     print library.clips[0].proxy_source()
+#
+#     print(args.input)
+#     print(args.save_file)
+
+if __name__ == '__main__':
+    library = FCPXLibrary('new.fcpxml')
+    # print library.clips[0].source
+    # print library.library_location()
+    #
+    # print dir(library.clips[0])
+    # print library.clips[0]
+    # print library.clips[0].proxy_source()
+    # for clip in library.clips:
+    #     print clip
+    print(dir(library.projects[0]))
+    print(library.formats)
+
+