29 | {% if screenshot not in (job_status.QUEUED, job_status.RUNNING, job_status.FAILED, None) %}
30 |
31 | {% elif screenshot == job_status.QUEUED %}
32 | Thumbnail generation in job queue
33 | {% elif screenshot == job_status.RUNNING %}
34 | Thumbnail generation running
35 | {% elif screenshot == job_status.FAILED %}
36 | Error occurred whilst generating thumbnail
37 | {% endif %}
38 |
61 | | Streams |
62 |
63 |
64 | {% if video_streams %}
65 | | Type | PID | Index | Height | Width | Display Aps. Ratio | Frame Rate | Codec | Bit Rate | Pixel Format | Start PTS |
|---|
66 | {% for video_stream_id in video_streams %}
67 |
68 | | {{ metadata['streams'][video_stream_id]['codec_type'] }} |
69 | {{ metadata['streams'][video_stream_id]['id'] or 'N/A' }} |
70 | {{ video_stream_id }} |
71 | {{ metadata['streams'][video_stream_id]['height'] or 'N/A' }} |
72 | {{ metadata['streams'][video_stream_id]['width'] or 'N/A' }} |
73 | {{ metadata['streams'][video_stream_id]['display_aspect_ratio'] or 'N/A' }} |
74 | {{ metadata['streams'][video_stream_id]['avg_frame_rate'] or 'N/A' }}
75 | ({{ metadata['streams'][video_stream_id]['avg_frame_rate_norm'] or 'N/A' }}) |
76 | {{ metadata['streams'][video_stream_id]['codec_name'] }} ({{ metadata['streams'][video_stream_id]['codec_long_name'] }}) |
77 | {{ metadata['streams'][video_stream_id]['bit_rate'] or 'N/A' }} |
78 | {{ metadata['streams'][video_stream_id]['pix_fmt'] or 'N/A' }} |
79 | {{ metadata['streams'][video_stream_id]['start_pts'] or 'N/A' }} |
80 |
81 | {% endfor %}
82 | {% else %}
83 | | No Video Streams available |
84 | {% endif %}
85 |
|
86 |
87 |
88 | {% if audio_streams %}
89 | | Type | PID | Index | Channels | Channel Layout | Sample Rate | Codec | Bit Rate | Bits per Sample | Sample Format |
|---|
90 | {% for audio_stream_id in audio_streams %}
91 |
92 | | {{ metadata['streams'][audio_stream_id]['codec_type'] }} |
93 | {{ metadata['streams'][audio_stream_id]['id'] or 'N/A' }} |
94 | {{ audio_stream_id }} |
95 | {{ metadata['streams'][audio_stream_id]['channels'] or 'N/A' }} |
96 | {{ metadata['streams'][audio_stream_id]['channel_layout'] or 'N/A' }} |
97 | {{ metadata['streams'][audio_stream_id]['sample_rate'] or 'N/A' }} |
98 | {{ metadata['streams'][audio_stream_id]['codec_name'] }} ({{ metadata['streams'][audio_stream_id]['codec_long_name'] }}) |
99 | {{ metadata['streams'][audio_stream_id]['bit_rate'] or 'N/A' }} |
100 | {{ metadata['streams'][audio_stream_id]['bits_per_sample'] or 'N/A' }} |
101 | {{ metadata['streams'][audio_stream_id]['sample_fmt'] or 'N/A' }} |
102 |
103 | {% endfor %}
104 | {% else %}
105 | | No Audio Streams available |
106 | {% endif %}
107 |
|
108 |
109 |
110 |
111 | {% if other_streams %}
112 | | Type | PID | Index | Codec | Bit Rate |
|---|
113 | {% for stream_id in other_streams %}
114 |
115 | | {{ metadata['streams'][stream_id]['codec_type'] }} |
116 | {{ metadata['streams'][stream_id]['id'] or 'N/A' }} |
117 | {{ stream_id }} |
118 | {{ metadata['streams'][stream_id]['codec_name'] }} ({{ metadata['streams'][stream_id]['codec_long_name'] }}) |
119 | {{ metadata['streams'][stream_id]['bit_rate'] or 'N/A' }} |
120 |
121 | {% endfor %}
122 | {% endif %}
123 | |
124 |
125 |
126 |
127 |
128 | {% elif metadata == job_status.QUEUED %}
129 | Metadata generation in job queue
130 | {% elif metadata == job_status.RUNNING %}
131 | Metadata generation running
132 | {% elif metadata == job_status.FAILED %}
133 | Error occurred whilst generating metadata
134 | {% endif %}
135 |
136 | {% endblock %}
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | GNU GENERAL PUBLIC LICENSE
2 | Version 2, June 1991
3 |
4 | Copyright (C) 1989, 1991 Free Software Foundation, Inc.,
5 | 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
6 | Everyone is permitted to copy and distribute verbatim copies
7 | of this license document, but changing it is not allowed.
8 |
9 | Preamble
10 |
11 | The licenses for most software are designed to take away your
12 | freedom to share and change it. By contrast, the GNU General Public
13 | License is intended to guarantee your freedom to share and change free
14 | software--to make sure the software is free for all its users. This
15 | General Public License applies to most of the Free Software
16 | Foundation's software and to any other program whose authors commit to
17 | using it. (Some other Free Software Foundation software is covered by
18 | the GNU Lesser General Public License instead.) You can apply it to
19 | your programs, too.
20 |
21 | When we speak of free software, we are referring to freedom, not
22 | price. Our General Public Licenses are designed to make sure that you
23 | have the freedom to distribute copies of free software (and charge for
24 | this service if you wish), that you receive source code or can get it
25 | if you want it, that you can change the software or use pieces of it
26 | in new free programs; and that you know you can do these things.
27 |
28 | To protect your rights, we need to make restrictions that forbid
29 | anyone to deny you these rights or to ask you to surrender the rights.
30 | These restrictions translate to certain responsibilities for you if you
31 | distribute copies of the software, or if you modify it.
32 |
33 | For example, if you distribute copies of such a program, whether
34 | gratis or for a fee, you must give the recipients all the rights that
35 | you have. You must make sure that they, too, receive or can get the
36 | source code. And you must show them these terms so they know their
37 | rights.
38 |
39 | We protect your rights with two steps: (1) copyright the software, and
40 | (2) offer you this license which gives you legal permission to copy,
41 | distribute and/or modify the software.
42 |
43 | Also, for each author's protection and ours, we want to make certain
44 | that everyone understands that there is no warranty for this free
45 | software. If the software is modified by someone else and passed on, we
46 | want its recipients to know that what they have is not the original, so
47 | that any problems introduced by others will not reflect on the original
48 | authors' reputations.
49 |
50 | Finally, any free program is threatened constantly by software
51 | patents. We wish to avoid the danger that redistributors of a free
52 | program will individually obtain patent licenses, in effect making the
53 | program proprietary. To prevent this, we have made it clear that any
54 | patent must be licensed for everyone's free use or not licensed at all.
55 |
56 | The precise terms and conditions for copying, distribution and
57 | modification follow.
58 |
59 | GNU GENERAL PUBLIC LICENSE
60 | TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
61 |
62 | 0. This License applies to any program or other work which contains
63 | a notice placed by the copyright holder saying it may be distributed
64 | under the terms of this General Public License. The "Program", below,
65 | refers to any such program or work, and a "work based on the Program"
66 | means either the Program or any derivative work under copyright law:
67 | that is to say, a work containing the Program or a portion of it,
68 | either verbatim or with modifications and/or translated into another
69 | language. (Hereinafter, translation is included without limitation in
70 | the term "modification".) Each licensee is addressed as "you".
71 |
72 | Activities other than copying, distribution and modification are not
73 | covered by this License; they are outside its scope. The act of
74 | running the Program is not restricted, and the output from the Program
75 | is covered only if its contents constitute a work based on the
76 | Program (independent of having been made by running the Program).
77 | Whether that is true depends on what the Program does.
78 |
79 | 1. You may copy and distribute verbatim copies of the Program's
80 | source code as you receive it, in any medium, provided that you
81 | conspicuously and appropriately publish on each copy an appropriate
82 | copyright notice and disclaimer of warranty; keep intact all the
83 | notices that refer to this License and to the absence of any warranty;
84 | and give any other recipients of the Program a copy of this License
85 | along with the Program.
86 |
87 | You may charge a fee for the physical act of transferring a copy, and
88 | you may at your option offer warranty protection in exchange for a fee.
89 |
90 | 2. You may modify your copy or copies of the Program or any portion
91 | of it, thus forming a work based on the Program, and copy and
92 | distribute such modifications or work under the terms of Section 1
93 | above, provided that you also meet all of these conditions:
94 |
95 | a) You must cause the modified files to carry prominent notices
96 | stating that you changed the files and the date of any change.
97 |
98 | b) You must cause any work that you distribute or publish, that in
99 | whole or in part contains or is derived from the Program or any
100 | part thereof, to be licensed as a whole at no charge to all third
101 | parties under the terms of this License.
102 |
103 | c) If the modified program normally reads commands interactively
104 | when run, you must cause it, when started running for such
105 | interactive use in the most ordinary way, to print or display an
106 | announcement including an appropriate copyright notice and a
107 | notice that there is no warranty (or else, saying that you provide
108 | a warranty) and that users may redistribute the program under
109 | these conditions, and telling the user how to view a copy of this
110 | License. (Exception: if the Program itself is interactive but
111 | does not normally print such an announcement, your work based on
112 | the Program is not required to print an announcement.)
113 |
114 | These requirements apply to the modified work as a whole. If
115 | identifiable sections of that work are not derived from the Program,
116 | and can be reasonably considered independent and separate works in
117 | themselves, then this License, and its terms, do not apply to those
118 | sections when you distribute them as separate works. But when you
119 | distribute the same sections as part of a whole which is a work based
120 | on the Program, the distribution of the whole must be on the terms of
121 | this License, whose permissions for other licensees extend to the
122 | entire whole, and thus to each and every part regardless of who wrote it.
123 |
124 | Thus, it is not the intent of this section to claim rights or contest
125 | your rights to work written entirely by you; rather, the intent is to
126 | exercise the right to control the distribution of derivative or
127 | collective works based on the Program.
128 |
129 | In addition, mere aggregation of another work not based on the Program
130 | with the Program (or with a work based on the Program) on a volume of
131 | a storage or distribution medium does not bring the other work under
132 | the scope of this License.
133 |
134 | 3. You may copy and distribute the Program (or a work based on it,
135 | under Section 2) in object code or executable form under the terms of
136 | Sections 1 and 2 above provided that you also do one of the following:
137 |
138 | a) Accompany it with the complete corresponding machine-readable
139 | source code, which must be distributed under the terms of Sections
140 | 1 and 2 above on a medium customarily used for software interchange; or,
141 |
142 | b) Accompany it with a written offer, valid for at least three
143 | years, to give any third party, for a charge no more than your
144 | cost of physically performing source distribution, a complete
145 | machine-readable copy of the corresponding source code, to be
146 | distributed under the terms of Sections 1 and 2 above on a medium
147 | customarily used for software interchange; or,
148 |
149 | c) Accompany it with the information you received as to the offer
150 | to distribute corresponding source code. (This alternative is
151 | allowed only for noncommercial distribution and only if you
152 | received the program in object code or executable form with such
153 | an offer, in accord with Subsection b above.)
154 |
155 | The source code for a work means the preferred form of the work for
156 | making modifications to it. For an executable work, complete source
157 | code means all the source code for all modules it contains, plus any
158 | associated interface definition files, plus the scripts used to
159 | control compilation and installation of the executable. However, as a
160 | special exception, the source code distributed need not include
161 | anything that is normally distributed (in either source or binary
162 | form) with the major components (compiler, kernel, and so on) of the
163 | operating system on which the executable runs, unless that component
164 | itself accompanies the executable.
165 |
166 | If distribution of executable or object code is made by offering
167 | access to copy from a designated place, then offering equivalent
168 | access to copy the source code from the same place counts as
169 | distribution of the source code, even though third parties are not
170 | compelled to copy the source along with the object code.
171 |
172 | 4. You may not copy, modify, sublicense, or distribute the Program
173 | except as expressly provided under this License. Any attempt
174 | otherwise to copy, modify, sublicense or distribute the Program is
175 | void, and will automatically terminate your rights under this License.
176 | However, parties who have received copies, or rights, from you under
177 | this License will not have their licenses terminated so long as such
178 | parties remain in full compliance.
179 |
180 | 5. You are not required to accept this License, since you have not
181 | signed it. However, nothing else grants you permission to modify or
182 | distribute the Program or its derivative works. These actions are
183 | prohibited by law if you do not accept this License. Therefore, by
184 | modifying or distributing the Program (or any work based on the
185 | Program), you indicate your acceptance of this License to do so, and
186 | all its terms and conditions for copying, distributing or modifying
187 | the Program or works based on it.
188 |
189 | 6. Each time you redistribute the Program (or any work based on the
190 | Program), the recipient automatically receives a license from the
191 | original licensor to copy, distribute or modify the Program subject to
192 | these terms and conditions. You may not impose any further
193 | restrictions on the recipients' exercise of the rights granted herein.
194 | You are not responsible for enforcing compliance by third parties to
195 | this License.
196 |
197 | 7. If, as a consequence of a court judgment or allegation of patent
198 | infringement or for any other reason (not limited to patent issues),
199 | conditions are imposed on you (whether by court order, agreement or
200 | otherwise) that contradict the conditions of this License, they do not
201 | excuse you from the conditions of this License. If you cannot
202 | distribute so as to satisfy simultaneously your obligations under this
203 | License and any other pertinent obligations, then as a consequence you
204 | may not distribute the Program at all. For example, if a patent
205 | license would not permit royalty-free redistribution of the Program by
206 | all those who receive copies directly or indirectly through you, then
207 | the only way you could satisfy both it and this License would be to
208 | refrain entirely from distribution of the Program.
209 |
210 | If any portion of this section is held invalid or unenforceable under
211 | any particular circumstance, the balance of the section is intended to
212 | apply and the section as a whole is intended to apply in other
213 | circumstances.
214 |
215 | It is not the purpose of this section to induce you to infringe any
216 | patents or other property right claims or to contest validity of any
217 | such claims; this section has the sole purpose of protecting the
218 | integrity of the free software distribution system, which is
219 | implemented by public license practices. Many people have made
220 | generous contributions to the wide range of software distributed
221 | through that system in reliance on consistent application of that
222 | system; it is up to the author/donor to decide if he or she is willing
223 | to distribute software through any other system and a licensee cannot
224 | impose that choice.
225 |
226 | This section is intended to make thoroughly clear what is believed to
227 | be a consequence of the rest of this License.
228 |
229 | 8. If the distribution and/or use of the Program is restricted in
230 | certain countries either by patents or by copyrighted interfaces, the
231 | original copyright holder who places the Program under this License
232 | may add an explicit geographical distribution limitation excluding
233 | those countries, so that distribution is permitted only in or among
234 | countries not thus excluded. In such case, this License incorporates
235 | the limitation as if written in the body of this License.
236 |
237 | 9. The Free Software Foundation may publish revised and/or new versions
238 | of the General Public License from time to time. Such new versions will
239 | be similar in spirit to the present version, but may differ in detail to
240 | address new problems or concerns.
241 |
242 | Each version is given a distinguishing version number. If the Program
243 | specifies a version number of this License which applies to it and "any
244 | later version", you have the option of following the terms and conditions
245 | either of that version or of any later version published by the Free
246 | Software Foundation. If the Program does not specify a version number of
247 | this License, you may choose any version ever published by the Free Software
248 | Foundation.
249 |
250 | 10. If you wish to incorporate parts of the Program into other free
251 | programs whose distribution conditions are different, write to the author
252 | to ask for permission. For software which is copyrighted by the Free
253 | Software Foundation, write to the Free Software Foundation; we sometimes
254 | make exceptions for this. Our decision will be guided by the two goals
255 | of preserving the free status of all derivatives of our free software and
256 | of promoting the sharing and reuse of software generally.
257 |
258 | NO WARRANTY
259 |
260 | 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY
261 | FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN
262 | OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
263 | PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED
264 | OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
265 | MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS
266 | TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE
267 | PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,
268 | REPAIR OR CORRECTION.
269 |
270 | 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
271 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR
272 | REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,
273 | INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING
274 | OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED
275 | TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY
276 | YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER
277 | PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE
278 | POSSIBILITY OF SUCH DAMAGES.
279 |
280 | END OF TERMS AND CONDITIONS
281 |
282 |
--------------------------------------------------------------------------------
/server.py:
--------------------------------------------------------------------------------
1 | #!/usr/bin/python
2 |
3 | # Copyright (c) 2016 - Matt Comben
4 | #
5 | # This file is part of TransportStreamBrowser.
6 | #
7 | # TransportStreamBrowser is free software: you can redistribute it and/or modify
8 | # it under the terms of the GNU General Public License as published by
9 | # the Free Software Foundation, either version 2 of the License, or
10 | # (at your option) any later version.
11 | #
12 | # TransportStreamBrowser is distributed in the hope that it will be useful,
13 | # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 | # GNU General Public License for more details.
16 | #
17 | # You should have received a copy of the GNU General Public License
18 |
19 | import os
20 | import subprocess
21 | import json
22 | import shutil
23 | import tempfile
24 | import time
25 | import threading
26 | import fnmatch
27 | from watchdog.events import FileSystemEventHandler
28 | from watchdog.observers import Observer
29 | from flask import Flask, render_template
30 | import glob
31 | import humanize
32 |
33 |
34 | TS_DIR = '/home/matthew/transportstream'
35 |
36 |
37 | class FileDoesNotExist(Exception):
38 | pass
39 |
40 |
41 | class InvalidFilename(Exception):
42 | pass
43 |
44 |
45 | class JobStatus(object):
46 | QUEUED = 1
47 | RUNNING = 2
48 | FAILED = 3
49 | COMPLETE = 4
50 |
51 |
52 | class JobType(object):
53 | METADATA = 1
54 | SCREENSHOT = 2
55 | SHUTDOWN = 3
56 |
57 |
58 | def run_command(command_args, cwd=None):
59 | """Run system command"""
60 | command_process = subprocess.Popen(
61 | command_args,
62 | stdout=subprocess.PIPE,
63 | stderr=subprocess.PIPE,
64 | cwd=cwd)
65 | rc = command_process.wait()
66 | stdout = command_process.stdout.read()
67 | stderr = command_process.stderr.read()
68 | return (rc, stdout, stderr)
69 |
70 |
71 | class Worker(object):
72 | QUEUE = []
73 | JOB_REFERENCE = {}
74 |
75 | @staticmethod
76 | def add_job(path, job_type, object_ref, start=False):
77 | key = (path, job_type)
78 | if key not in Worker.QUEUE:
79 | if start:
80 | Worker.QUEUE.insert(0, key)
81 | else:
82 | Worker.QUEUE.append(key)
83 | Worker.JOB_REFERENCE[key] = object_ref
84 | object_ref.job_status[job_type] = JobStatus.QUEUED
85 |
86 | def __init__(self):
87 | self.active = True
88 | self.thread_ref = threading.Thread(target=self.run)
89 |
90 | def start(self):
91 | self.thread_ref.start()
92 |
93 | def run(self):
94 | while self.active:
95 | if not len(Worker.QUEUE):
96 | time.sleep(2)
97 | else:
98 | job = Worker.QUEUE[0]
99 | if job[1] == JobType.METADATA:
100 | Worker.JOB_REFERENCE[job].generate_metadata()
101 | elif job[1] == JobType.SCREENSHOT:
102 | Worker.JOB_REFERENCE[job].generate_screenshot()
103 | elif job[1] == JobType.SHUTDOWN:
104 | self.active = False
105 | Worker.QUEUE.remove(job)
106 | del(Worker.JOB_REFERENCE[job])
107 |
108 | def shutdown(self):
109 | self.active = False
110 | class Fake(object):
111 | def __init__(self):
112 | self.job_status = {'shutdown': 0}
113 | Worker.add_job('shutdown', JobType.SHUTDOWN, Fake(), start=True)
114 |
115 |
116 | class Inspector(object):
117 |
118 | CACHED_OBJECTS = {}
119 |
120 | @staticmethod
121 | def get_inspector(filename):
122 | if filename not in Inspector.CACHED_OBJECTS:
123 | Inspector.CACHED_OBJECTS[filename] = Inspector(filename)
124 | return Inspector.CACHED_OBJECTS[filename]
125 |
126 | def __init__(self, filename):
127 | self.filename = filename
128 | self.job_status = {
129 | JobType.SCREENSHOT: None,
130 | JobType.METADATA: None
131 | }
132 |
133 | def get_file_paths(self):
134 | if not os.path.isfile(self.filename):
135 | raise FileDoesNotExist('%s does not exist' % self.filename)
136 |
137 | filename = os.path.basename(self.filename)
138 | if filename.startswith('.'):
139 | raise InvalidFilename('%s starts with invalid character' % self.filename)
140 | if not filename.endswith('.ts'):
141 | raise InvalidFilename('%s does not end in \'ts\'' % self.filename)
142 |
143 | directory = os.path.dirname(self.filename)
144 | filename = os.path.basename(self.filename)
145 | web_dir = directory.replace(TS_DIR, '')
146 | if not web_dir.startswith('/'):
147 | web_dir = "/%s" % web_dir
148 | return {
149 | 'dir': directory,
150 | 'filename': filename,
151 | 'metadata_file': '%s/.%s-meta' % (directory, filename),
152 | 'screenshot': '%s/.%s-screenshot' % (directory, filename),
153 | 'web_path': self.filename.replace(TS_DIR, ''),
154 | 'web_dir': web_dir
155 | }
156 |
157 | def get_screenshot(self, force_update=False):
158 | paths = self.get_file_paths()
159 |
160 | if force_update or not os.path.isfile(paths['screenshot']):
161 | if force_update or self.job_status[JobType.SCREENSHOT] != JobStatus.FAILED:
162 | Worker.add_job(self.filename, JobType.SCREENSHOT, self)
163 | return self.job_status[JobType.SCREENSHOT]
164 | else:
165 | with open(paths['screenshot'], 'rb') as fh:
166 | data = fh.read()
167 | return data.encode('base64')
168 |
169 | def get_metadata(self, force_update=False):
170 | paths = self.get_file_paths()
171 | if force_update or not os.path.isfile(paths['metadata_file']):
172 | if force_update or self.job_status[JobType.METADATA] != JobStatus.FAILED:
173 | Worker.add_job(self.filename, JobType.METADATA, self)
174 | return self.job_status[JobType.METADATA]
175 | else:
176 | with open(paths['metadata_file'], 'r') as fh:
177 | data = json.loads(fh.read())
178 | return data
179 |
180 | def generate_metadata(self):
181 | try:
182 | self.job_status[JobType.METADATA] = JobStatus.RUNNING
183 | rc, output, stderr = run_command(['ffprobe', '-show_streams', '-show_format',
184 | '-show_chapters', '-show_programs', '-v', 'quiet',
185 | '-print_format', 'json', self.filename])
186 | assert (rc == 0)
187 |
188 | metadata = json.loads(output)
189 | paths = self.get_file_paths()
190 | with open(paths['metadata_file'], 'w') as fh:
191 | fh.write(json.dumps(metadata))
192 | self.job_status[JobType.METADATA] = JobStatus.COMPLETE
193 | except:
194 | self.job_status[JobType.METADATA] = JobStatus.FAILED
195 |
196 | def generate_screenshot(self, force_update=False):
197 | temp_dir = tempfile.mkdtemp()
198 | try:
199 | self.job_status[JobType.SCREENSHOT] = JobStatus.RUNNING
200 | rc, output, stderr = run_command(['cvlc', self.filename, '--rate', '1',
201 | '--video-filter', 'scene', '--vout',
202 | 'dummy', '--start-time', '0', '--stop-time', '1',
203 | '--scene-format', 'png', '--scene-ratio', '10000000',
204 | '--scene-prefix', 'snap', '--scene-path', temp_dir,
205 | '--play-and-exit'])
206 | assert (rc == 0)
207 | assert len(os.listdir(temp_dir))
208 | src_screenshot = '%s/%s' % (temp_dir, os.listdir(temp_dir)[0])
209 | paths = self.get_file_paths()
210 | shutil.copyfile(src_screenshot, paths['screenshot'])
211 | assert os.path.isfile(paths['screenshot'])
212 | self.job_status[JobType.SCREENSHOT] = JobStatus.COMPLETE
213 | shutil.rmtree(temp_dir)
214 | except:
215 | self.job_status[JobType.SCREENSHOT] = JobStatus.FAILED
216 | shutil.rmtree(temp_dir)
217 | raise
218 |
219 |
220 |
221 | class Searcher(object):
222 |
223 | def get_file_list(self):
224 | matches = []
225 | for root, dirnames, filenames in os.walk(TS_DIR):
226 | for filename in fnmatch.filter(filenames, '*.ts'):
227 | matches.append(os.path.join(root, filename))
228 | return matches
229 |
230 | def get_all_files_in_dir(self, directory):
231 | inspectors = {}
232 | for file_name in glob.glob('%s/*.ts' % directory):
233 | try:
234 | inspectors[file_name] = Inspector.get_inspector(file_name)
235 | except:
236 | pass
237 | return inspectors
238 |
239 | def get_all_dirs_in_dir(self, directory):
240 | directories = []
241 | for file_name in glob.glob('%s/*' % directory):
242 | if os.path.isdir(file_name) and file_name not in ('.', '..'):
243 | directories.append(file_name)
244 | return directories
245 |
246 | def get_all_inspector_objects(self):
247 | inspector_objects = {}
248 | for filename in self.get_file_list():
249 | inspector_objects[filename] = Inspector.get_inspector(filename)
250 | return inspector_objects
251 |
252 | class MonitorHandler(FileSystemEventHandler):
253 | def on_deleted(self, event):
254 | if not event.is_directory:
255 | inspect_obj = Inspector.get_inspector(event.src_path)
256 | paths = inspect_obj.get_file_paths()
257 | if os.path.exists(paths['metadata_file']):
258 | os.unlink(paths['metadata_file'])
259 |
260 | if os.path.exists(paths['screenshot']):
261 | os.unlink(paths['screenshot'])
262 |
263 | def on_modified(self, event):
264 | if not event.is_directory:
265 | inspect_obj = Inspector.get_inspector(event.src_path)
266 | paths = inspect_obj.get_file_paths()
267 | if os.path.exists(paths['metadata_file']):
268 | os.unlink(paths['metadata_file'])
269 |
270 | if os.path.exists(paths['screenshot']):
271 | os.unlink(paths['screenshot'])
272 | inspect_obj.get_screenshot()
273 | inspect_obj.get_metadata()
274 |
275 | def on_moved(self, event):
276 | if not event.is_directory:
277 | inspect_obj = Inspector.get_inspector(event.dest_path)
278 | paths = inspect_obj.get_file_paths()
279 | if os.path.exists(paths['metadata_file']):
280 | os.unlink(paths['metadata_file'])
281 |
282 | if os.path.exists(paths['screenshot']):
283 | os.unlink(paths['screenshot'])
284 | inspect_obj.get_screenshot()
285 | inspect_obj.get_metadata()
286 |
287 | def on_created(self, event):
288 | if not event.is_directory:
289 | inspect_obj = Inspector.get_inspector(event.src_path)
290 | paths = inspect_obj.get_file_paths()
291 | if os.path.exists(paths['metadata_file']):
292 | os.unlink(paths['metadata_file'])
293 |
294 | if os.path.exists(paths['screenshot']):
295 | os.unlink(paths['screenshot'])
296 | inspect_obj.get_screenshot()
297 | inspect_obj.get_metadata()
298 |
299 |
300 | class Monitor(object):
301 |
302 | def __init__(self):
303 | self.observer = Observer()
304 | handler = MonitorHandler()
305 | self.observer.schedule(handler, TS_DIR, recursive=True)
306 |
307 | def start(self):
308 | self.observer.start()
309 |
310 | def shutdown(self):
311 | self.observer.stop()
312 |
313 | class WebServer(object):
314 | def __init__(self):
315 | self.app = Flask(__name__)
316 |
317 | @self.app.route('/', defaults={'path': './'})
318 | @self.app.route('/
30 | Transport Stream Browser
31 |