moparisthebest/HttpUploadComponent
1
0
Fork 0
mirror of https://github.com/moparisthebest/HttpUploadComponent synced 2024-12-22 07:28:51 -05:00
Code Issues Releases Wiki Activity

Merge pull request #22 from chaotix-/mime-content-disposition

Browse Source 
Just send Content-disposition: attachment-Header if content is no image
This commit is contained in:
Daniel Gultsch 2015-07-27 13:16:54 +02:00
commit 2f0fec6c7f
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@ -8,16 +8,12 @@ import mimetypes
import os import os
import random import random
import shutil import shutil
import signal
import sleekxmpp
import ssl import ssl
import string import string
import sys
import yaml import yaml
from sleekxmpp.componentxmpp import ComponentXMPP from sleekxmpp.componentxmpp import ComponentXMPP
from threading import Lock from threading import Lock
from threading import Thread
try: try:
# Python 3 # Python 3
@ -150,7 +146,8 @@ class HttpHandler(BaseHTTPRequestHandler):
if mime is None: if mime is None:
mime = 'application/octet-stream' mime = 'application/octet-stream'
self.send_header("Content-Type", mime) self.send_header("Content-Type", mime)
self.send_header("Content-Disposition", 'attachment; filename="{}"'.format(os.path.basename(filename))) if mime[:6] != 'image/':
self.send_header("Content-Disposition", 'attachment; filename="{}"'.format(os.path.basename(filename)))
fs = os.fstat(f.fileno()) fs = os.fstat(f.fileno())
self.send_header("Content-Length", str(fs.st_size)) self.send_header("Content-Length", str(fs.st_size))
self.end_headers() self.end_headers()
@ -174,7 +171,7 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-c", "--config", default='config.yml', help='Specify alternate config file.') parser.add_argument("-c", "--config", default='config.yml', help='Specify alternate config file.')
parser.add_argument("-l", "--logfile", default=None, help='File where the server log will be stored. If not specified log to stdout.') parser.add_argument("-l", "--logfile", default=None, help='File where the server log will be stored. If not specified log to stdout.')
args = parser.parse_args() args = parser.parse_args()
with open(args.config,'r') as ymlfile: with open(args.config,'r') as ymlfile: