Created
August 13, 2011 12:26
-
-
Save russss/1143799 to your computer and use it in GitHub Desktop.
Grab a single frame from an MJPEG stream
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
import socket | |
import sys | |
if len(sys.argv) != 3: | |
print "Usage: %s host:port destfile.jpg" % sys.argv[0] | |
sys.exit(1) | |
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
host, port = sys.argv[1].split(':') | |
s.connect((host, int(port))) | |
fh = s.makefile() | |
# Read in HTTP headers: | |
line = fh.readline() | |
while line.strip() != '': | |
parts = line.split(':') | |
if len(parts) > 1 and parts[0].lower() == 'content-type': | |
# Extract boundary string from content-type | |
content_type = parts[1].strip() | |
boundary = content_type.split(';')[1].split('=')[1] | |
line = fh.readline() | |
if not boundary: | |
raise Exception("Can't find content-type") | |
# Seek ahead to the first chunk | |
while line.strip() != boundary: | |
line = fh.readline() | |
# Read in chunk headers | |
while line.strip() != '': | |
parts = line.split(':') | |
if len(parts) > 1 and parts[0].lower() == 'content-length': | |
# Grab chunk length | |
length = int(parts[1].strip()) | |
line = fh.readline() | |
image = fh.read(length) | |
with open(sys.argv[2], 'w') as out_fh: | |
out_fh.write(image) | |
s.close() |
This worked fine for years until I upgraded to Ubuntu 18.04 LTS where it suddenly no longer worked. Making the change suggested by jogu got it back to working again. Thanks, jogu.
Thanks, works great!
I kept the
# Seek ahead to the first chunk
section as is.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thank you so much, very useful!