Mercurialgmyth / file revision
summary | shortlog | changelog | graph | tags | branches | files | changeset | file | revisions | annotate | diff | raw
gmyth-stream/server/plugins/transcoders/mencoder_lib/mythtv.py
author melunko
Mon Mar 03 11:33:49 2008 +0000 (2008-03-03)
branchtrunk
changeset 935 067904b6921f
parent 765 57f3de326cd8
permissions -rw-r--r--
[svn r944] Now add/cancel a schedule recording while livetv is running is finnaly working fine. 
     1 import os

     2 import subprocess

     3 import fcntl

     4 

     5 import lib.utils as utils

     6 import lib.server as server

     7 

     8 from select import select

     9 

    10 def _setup_mythfilename(self):

    11     # mythtv:mythtv@192.168.3.110:6543/1002_20070426230000.nuv

    12     try:

    13         _mysql = self.args["input"].split("@")[0].split(":")

    14     except IndexError, e:

    15         _mysql = ["mythtv", "mythtv"]

    16 

    17     try:

    18         _args = self.args["input"].split("@")[1].split(":")

    19     except IndexError, e:

    20         _args = self.args["input"].split(":")

    21 

    22     gmyth_dict = {}

    23     gmyth_dict["mysql"] = _mysql

    24     gmyth_dict["backend"] = _args[0]

    25     gmyth_dict["port"] = _args[1].split("/", 1)[0]

    26 

    27     _tmp_file = _args[1].split("/", 1)[1]

    28 

    29     if _tmp_file.find("channel") >= 0:

    30         gmyth_dict["kind"] = "c"

    31         gmyth_dict["cfile"] = _tmp_file.split("=")[1]

    32     else:

    33         gmyth_dict["kind"] = "f"

    34         gmyth_dict["cfile"] = _tmp_file

    35 

    36     self.args["input"] = "-"

    37     return gmyth_dict

    38 # _setup_mythfilename

    39 

    40 def _setup_mythfile(err):

    41     size = err.readline().split("Size:")[1]

    42     flags = fcntl.fcntl (err, fcntl.F_GETFL, 0) | os.O_NONBLOCK

    43     fcntl.fcntl(err, fcntl.F_SETFL, flags)

    44     return size

    45 # _setup_mythfile

    46 

    47 def _setup_gmythcat(self):

    48     gmyth_cat = utils.which("gmyth-cat")

    49     if self.args.has_key("gmyth-cat"):

    50         return [ utils.which("gmyth-cat"),

    51                  "-h", self.args["gmyth-cat"]["backend"],

    52                  "-p", self.args["gmyth-cat"]["port"],

    53                  "-" + self.args["gmyth-cat"]["kind"],

    54                  self.args["gmyth-cat"]["cfile"]

    55                  ]

    56     else:

    57         self.log.error(self.tid, "Error: URI error")

    58         return []

    59 # _setup_gmythcat

    60 

    61 def start_myth(self, outfd):

    62     opts = _setup_gmythcat(self)

    63     try:

    64         self.gmyth = subprocess.Popen(opts, stdout=subprocess.PIPE,

    65                                       stderr=subprocess.PIPE,

    66                                       close_fds=True)

    67     except Exception, e:

    68         self.log.error(self.tid, "Error: gmyth-cat: %s" % e)

    69         return False

    70 

    71     if not self._run_mencoder(input=self.gmyth.stdout,

    72                               output=subprocess.PIPE):

    73         return False

    74 

    75     if self.args["gmyth-cat"]["kind"] == "f":

    76         try:

    77             size = _setup_mythfile(self.gmyth.stderr)

    78             self.log.debug(self.tid, "Info: Size of file: %s" % size)

    79         except Exception, e:

    80             self.log.error(self.tid, "Error: Problems getting size of"\

    81                            " file: %s" % e)

    82             outfd.write("Error: Problems getting size of file: %s" % e)

    83             return False

    84 

    85     outfd.write("OK     ")

    86 

    87     try:

    88         while self.proc and self.proc.poll() == None:

    89             r, w, x = select([self.gmyth.stderr, self.proc.stdout],

    90                              [], [], 0)

    91             if self.proc.stdout in r:

    92                 d = self.proc.stdout.read(4096)

    93                 outfd.write(d)

    94 

    95             if self.gmyth.stderr in r:

    96                 partial = self.gmyth.stderr.readline()

    97                 if partial != "":

    98                     self.status = utils.progress_bar(int(partial),

    99                                                      int(size), 50)

   100 

   101     except IndexError, e:

   102         pass

   103     except Exception, e:

   104         self.log.error(self.tid, "Error: %s" % e)

   105         return False

   106 

   107     self.log.info(self.tid, "OK: Done")

   108     return True

   109 # _start_myth()
gmyth