Ticket #3655: mythburn.py.patch

mytharchive/mythburn/scripts/mythburn.py

@@ -14 +14 @@
 # paulh
 # 4 May 2006 - Added into mythtv svn
 
+# robg
+# 20 Jun 2007 - Added commercial skipping chapter marks
+
 #For this script to work you need to have...
 #Python2.3.5
 #python2.3-mysqldb
@@ -31 +34 @@
 #******************************************************************************
 
 # version of script - change after each update
-VERSION="0.1.20070619-1"
+VERSION="0.1.20070620-2"
 
 
 ##You can use this debug flag when testing out new themes
@@ -522 +525 @@
 
     write( "Video length is %s seconds. Each chapter will be %s seconds" % (lengthofvideo,segment))
 
-    chapters=""
+    chapters = createVideoChaptersFixedLength(itemnum, segment, lengthofvideo)
+
     thumbList=""
     starttime=0
     count=1
     while count<=numofchapters:
-        chapters+=time.strftime("%H:%M:%S",time.gmtime(starttime))
-
         if starttime==0:
             if thumboffset < segment:
                 thumbList+="%s," % thumboffset
@@ -537 +539 @@
         else:
             thumbList+="%s," % starttime
 
-        if numofchapters>1:
-            chapters+="," 
         starttime+=segment
         count+=1
 
@@ -548 +548 @@
 
     return chapters
 
-def createVideoChaptersFixedLength(segment, lengthofvideo): 
-    """Returns chapter marks spaced segment seconds through the file"""
+def createVideoChaptersFixedLength(itemnum, segment, lengthofvideo): 
+    """Returns chapter marks at commercial ends, or spaced segment seconds through the file"""
+
+    # if there is a chapter list already available, use it
+    infoDOM = xml.dom.minidom.parse(os.path.join(getItemTempPath(itemnum),"info.xml"))
+    chapterlistNode = infoDOM.getElementsByTagName("chapterlist")
+    if chapterlistNode.length > 0:
+        chapterlist = getText(chapterlistNode[0])
+        write("Using commercial end marks - %s" % chapterlist)
+        return chapterlist
+
     if lengthofvideo < segment:
         return "00:00:00"
 
@@ -871 +880 @@
         node.appendChild(infoDOM.createTextNode(""))
         top_element.appendChild(node)
 
-        #if this a myth recording we still need to find the chanid, starttime and hascutlist
+        #if this a myth recording we still need to find the chanid, starttime, hascutlist, and commercial end marks
         if file.attributes["type"].value=="recording":
             basename = os.path.basename(file.attributes["filename"].value)
             sqlstatement  = """SELECT starttime, chanid FROM recorded 
@@ -918 +927 @@
                     node.appendChild(infoDOM.createTextNode("no"))
                     top_element.appendChild(node)
 
+                # find the commercial end marks if available
+                sqlstatement  = """SELECT mark, type FROM recordedmarkup 
+                                WHERE chanid = '%s' AND starttime = '%s' 
+                                AND type = 5 ORDER BY mark""" % (chanid, starttime)
+                cursor = db.cursor()
+                # execute SQL statement
+                cursor.execute(sqlstatement)
+                # get the resultset as a tuple
+                result = cursor.fetchall()
+                if cursor.rowcount > 0:
+                    res, fps, ar = getVideoParams(folder)
+                    chapterlist="00:00:00"
+                    #iterate through marks, adding to chapterlist
+                    for record in result:
+                        chapterlist += "," + frameToTime(int(record[0]), float(fps))
+        
+                    node = infoDOM.createElement("chapterlist")
+                    node.appendChild(infoDOM.createTextNode(chapterlist))
+                    top_element.appendChild(node)
+
                 db.close()
                 del db
                 del cursor
@@ -1016 +1045 @@
                 node.appendChild(infoDOM.createTextNode("no"))
                 top_element.appendChild(node)
 
+            # find the commercial end marks if available
+            sqlstatement  = """SELECT mark, type FROM recordedmarkup 
+                                WHERE chanid = '%s' AND starttime = '%s' 
+                                AND type = 5 ORDER BY mark""" % (chanid, starttime)
+            cursor = db.cursor()
+            # execute SQL statement
+            cursor.execute(sqlstatement)
+            # get the resultset as a tuple
+            result = cursor.fetchall()
+            if cursor.rowcount > 0:
+                res, fps, ar = getVideoParams(folder)
+                chapterlist="00:00:00"
+                #iterate through marks, adding to chapterlist
+                for record in result:
+                    chapterlist += "," + frameToTime(int(record[0]), float(fps))
+        
+                node = infoDOM.createElement("chapterlist")
+                node.appendChild(infoDOM.createTextNode(chapterlist))
+                top_element.appendChild(node)
+
         db.close()
         del db
         del cursor
@@ -2066 +2115 @@
             if wantChapterMenu:
                 vob.setAttribute("chapters",createVideoChapters(itemnum,chapters,getLengthOfVideo(itemnum),False) )
             else:
-                vob.setAttribute("chapters", createVideoChaptersFixedLength(chapterLength, getLengthOfVideo(itemnum)))
+                vob.setAttribute("chapters", createVideoChaptersFixedLength(itemnum,chapterLength, getLengthOfVideo(itemnum)))
 
             vob.setAttribute("file",os.path.join(getItemTempPath(itemnum),"final.mpg"))
             pgc.appendChild(vob)
@@ -2284 +2333 @@
 
         vob = dvddom.createElement("vob")
         vob.setAttribute("file", os.path.join(getItemTempPath(itemNum), "final.mpg"))
-        vob.setAttribute("chapters", createVideoChaptersFixedLength(chapterLength, getLengthOfVideo(itemNum)))
+        vob.setAttribute("chapters", createVideoChaptersFixedLength(itemNum, chapterLength, getLengthOfVideo(itemNum)))
         pgc.appendChild(vob)
 
         del vob