ubuntu11.04(unix 就可以了) 共享文件以及支持上傳文件

時間 2021-01-02

我這兩個功能是分開實現的。。。

共享文件：

很簡單，就一條命令:

python -m SimpleHTTPServer

如果要指定端口號直接在剛纔的命令後面加上你要的端口號。接着別人就可以用你的IP訪問了。（在哪個目錄下使用這條命令就共享該文件夾）

支持上傳文件：

首先建一個droop文件，寫入以下代碼（代碼最後才列出），然後放到~/bin/目錄下，接着敲下面的命令：

mkdir ~/uploads

cd ~/uploads

python -m "Hi, I am alfred! Here you can upload you file to me:)" -p ~/bin/mypicture

這樣人家就可以根據你的ip訪問了，以下是截圖

droop的代碼：

#!/usr/bin/env python # -*- coding: utf-8 -*- # Droopy (http://stackp.online.fr/droopy) # Copyright 2008-2011 (c) Pierre Duquesne <[email protected]> # Licensed under the New BSD License. # Changelog # 20110708 * Polish translation by Jacek Politowski. # 20110625 * Fix bug regarding filesystem name encoding. # * Save the --dl option when --save-config is passed. # 20110501 * Add the --dl option to let clients download files. # * CSS speech bubble. # 20101130 * CSS and HTML update. Switch to the new BSD License. # 20100523 * Simplified Chinese translation by Ye Wei. # 20100521 * Hungarian translation by Csaba Szigetv謾ri. # * Russian translation by muromec. # * Use %APPDATA% Windows environment variable -- fix by Maik. # 20091229 * Brazilian Portuguese translation by # Carlos Eduardo Moreira dos Santos and Toony Poony. # * IE layout fix by Carlos Eduardo Moreira dos Santos. # * Galician translation by Miguel Anxo Bouzada. # 20090721 * Indonesian translation by Kemas. # 20090205 * Japanese translation by Satoru Matsumoto. # * Slovak translation by CyberBoBaK. # 20090203 * Norwegian translation by Preben Olav Pedersen. # 20090202 * Korean translation by xissy. # * Fix for unicode filenames by xissy. # * Relies on 127.0.0.1 instead of "localhost" hostname. # 20090129 * Serbian translation by kotnik. # 20090125 * Danish translation by jan. # 20081210 * Greek translation by n2j3. # 20081128 * Slovene translation by david. # * Romanian translation by Licaon. # 20081022 * Swedish translation by David Eurenius. # 20081001 * Droopy gets pretty (css and html rework). # * Finnish translation by ipppe. # 20080926 * Configuration saving and loading. # 20080906 * Extract the file base name (some browsers send the full path). # 20080905 * File is uploaded directly into the specified directory. # 20080904 * Arabic translation by Djalel Chefrour. # * Italian translation by fabius and d1s4st3r. # * Dutch translation by Tonio Voerman. # * Portuguese translation by Pedro Palma. # * Turkish translation by Heartsmagic. # 20080727 * Spanish translation by Federico Kereki. # 20080624 * Option -d or --directory to specify the upload directory. # 20080622 * File numbering to avoid overwriting. # 20080620 * Czech translation by Ji艡鉚. # * German translation by Michael. # 20080408 * First release. import BaseHTTPServer import SocketServer import cgi import os import posixpath import macpath import ntpath import sys import getopt import mimetypes import copy import shutil import tempfile import socket import locale import urllib LOGO = '''\ _____ | \.----.-----.-----.-----.--.--. | -- | _| _ | _ | _ | | | |_____/|__| |_____|_____| __|___ | |__| |_____| ''' USAGE='''\ Usage: droopy [options] [PORT] Options: -h, --help show this help message and exit -d DIRECTORY, --directory DIRECTORY set the directory to upload files to -m MESSAGE, --message MESSAGE set the message -p PICTURE, --picture PICTURE set the picture --dl provide download links --save-config save options in a configuration file --delete-config delete the configuration file and exit Example: droopy -m "Hi, this is Bob. You can send me a file." -p avatar.png ''' picture = None message = "" port = 8000 directory = os.curdir must_save_options = False publish_files = False # -- HTML templates style = '''<style type="text/css"> </style>''' userinfo = ''' <div id="userinfo"> %(message)s %(divpicture)s </div> ''' maintmpl = '''<html><head><title>%(maintitle)s</title> ''' + style + ''' <script language="JavaScript"> function swap() { document.getElementById("form").style.display = "none"; document.getElementById("sending").style.display = "block"; update(); } ncell = 4; curcell = 0; function update() { setTimeout(update, 300); e = document.getElementById("cell"+curcell); e.style.backgroundColor = "#fff"; curcell = (curcell+1) %% ncell e = document.getElementById("cell"+curcell); e.style.backgroundColor = "#369"; } function onunload() { document.getElementById("form").style.display = "block"; document.getElementById("sending").style.display = "none"; } </script></head> <body> %(linkurl)s <div id="wrapform"> <div id="form" class="box"> <form method="post" enctype="multipart/form-data" action=""> <input name="upfile" type="file"> <input value="%(submit)s" onclick="swap()" type="submit"> </form> </div> <div id="sending" class="box"> %(sending)s <table id="progress"><tr> <td id="cell0"/><td id="cell1"/><td id="cell2"/><td id="cell3"/> </tr></table> </div> </div> ''' + userinfo + ''' %(files)s </body></html> ''' successtmpl = ''' <html> <head><title> %(successtitle)s </title> ''' + style + ''' </head> <body> <div id="wrapform"> <div class="box"> %(received)s <a href="/"> %(another)s </a> </div> </div> ''' + userinfo + ''' </body> </html> ''' errortmpl = ''' <html> <head><title> %(errortitle)s </title> ''' + style + ''' </head> <body> <div id="wrapform"> <div class="box"> %(problem)s <a href="/"> %(retry)s </a> </div> </div> ''' + userinfo + ''' </body> </html> ''' linkurltmpl = '''<div id="linkurl" class="box"> <a href="http://stackp.online.fr/droopy-ip.php?port=%(port)d"> %(discover)s </a></div>''' templates = {"main": maintmpl, "success": successtmpl, "error": errortmpl} # -- Translations ar = {"maintitle": u"廿乇爻丕賱賲賱賮", "submit": u"廿乇爻丕賱", "sending": u"丕賱賲賱賮賯賷丿丕賱廿乇爻丕賱", "successtitle": u"鬲賲丕爻鬲賯亙丕賱丕賱賲賱賮", "received": u"鬲賲丕爻鬲賯亙丕賱丕賱賲賱賮 !", "another": u"廿乇爻丕賱賲賱賮丌禺乇", "errortitle": u"賲卮賰賱丞", "problem": u"丨丿孬鬲賲卮賰賱丞 !", "retry": u"廿毓丕丿丞丕賱賲丨丕賵賱丞", "discover": u"丕賰鬲卮丕賮毓賳賵丕賳賴匕賴丕賱氐賮丨丞"} cs = {"maintitle": u"Poslat soubor", "submit": u"Poslat", "sending": u"Pos鉚l謾m", "successtitle": u"Soubor doru膷en", "received": u"Soubor doru膷en !", "another": u"Poslat dal擰鉚 soubor", "errortitle": u"Chyba", "problem": u"Stala se chyba !", "retry": u"Zkusit znova.", "discover": u"Zjistit adresu str謾nky"} da = {"maintitle": u"Send en fil", "submit": u"Send", "sending": u"Sender", "successtitle": u"Fil modtaget", "received": u"Fil modtaget!", "another": u"Send en fil til.", "errortitle": u"Problem", "problem": u"Det er opst氓et en fejl!", "retry": u"Fors酶g igen.", "discover": u"Find adressen til denne side"} de = {"maintitle": "Datei senden", "submit": "Senden", "sending": "Sendet", "successtitle": "Datei empfangen", "received": "Datei empfangen!", "another": "Weitere Datei senden", "errortitle": "Fehler", "problem": "Ein Fehler ist aufgetreten!", "retry": "Wiederholen", "discover": "Internet-Adresse dieser Seite feststellen"} el = {"maintitle": u"危蟿蔚委位蔚葦謂僞僞蟻蠂蔚委慰", "submit": u"螒蟺慰蟽蟿慰位萎", "sending": u"螒蟺慰蟽蟿葦位位蔚蟿僞喂...", "successtitle": u"螘蟺喂蟿蠀蠂萎蟼位萎蠄畏僞蟻蠂蔚委慰蠀 ", "received": u"螞萎蠄畏僞蟻蠂蔚委慰蠀慰位慰魏位畏蟻蠋胃畏魏蔚", "another": u"危蟿蔚委位蔚維位位慰葦謂僞僞蟻蠂蔚委慰", "errortitle": u"危蠁維位渭僞", "problem": u"螤僞蟻慰蠀蟽喂維蟽蟿畏魏蔚蟽蠁維位渭僞", "retry": u"螘蟺僞謂維位畏蠄畏", "discover": u"螔蟻蔚蟼蟿畏謂未喂蔚蠉胃蠀謂蟽畏蟿畏蟼蟽蔚位委未僞蟼"} en = {"maintitle": "Send a file", "submit": "Send", "sending": "Sending", "successtitle": "File received", "received": "File received !", "another": "Send another file.", "errortitle": "Problem", "problem": "There has been a problem !", "retry": "Retry.", "discover": "Discover the address of this page"} es = {"maintitle": u"Enviar un archivo", "submit": u"Enviar", "sending": u"Enviando", "successtitle": u"Archivo recibido", "received": u"隆Archivo recibido!", "another": u"Enviar otro archivo.", "errortitle": u"Error", "problem": u"隆Hubo un problema!", "retry": u"Reintentar", "discover": u"Descubrir la direcci貿n de esta p謾gina"} fi = {"maintitle": u"L盲het盲 tiedosto", "submit": u"L盲het盲", "sending": u"L盲hett盲盲", "successtitle": u"Tiedosto vastaanotettu", "received": u"Tiedosto vastaanotettu!", "another": u"L盲het盲 toinen tiedosto.", "errortitle": u"Virhe", "problem": u"Virhe lahetett盲ess盲 tiedostoa!", "retry": u"Uudelleen.", "discover": u"N盲yt盲 t盲m盲n sivun osoite"} fr = {"maintitle": u"Envoyer un fichier", "submit": u"Envoyer", "sending": u"Envoi en cours", "successtitle": u"Fichier re莽u", "received": u"Fichier re莽u !", "another": u"Envoyer un autre fichier.", "errortitle": u"Probl貓me", "problem": u"Il y a eu un probl貓me !", "retry": u"R茅essayer.", "discover": u"D茅couvrir l'adresse de cette page"} gl = {"maintitle": u"Enviar un ficheiro", "submit": u"Enviar", "sending": u"Enviando", "successtitle": u"Ficheiro recibido", "received": u"Ficheiro recibido!", "another": u"Enviar outro ficheiro.", "errortitle": u"Erro", "problem": u"Xurd鉚u un problema!", "retry": u"Reintentar", "discover": u"Descubrir o enderezo desta p謾xina"} hu = {"maintitle": u"脕llom謾ny k眉ld茅se", "submit": u"K眉ld茅s", "sending": u"K眉ld茅s folyamatban", "successtitle": u"Az 謾llom謾ny be茅rkezett", "received": u"Az 謾llom謾ny be茅rkezett!", "another": u"Tov謾bbi 謾llom謾nyok k眉ld茅se", "errortitle": u"Hiba", "problem": u"Egy hiba l茅pett fel!", "retry": u"Megism茅telni", "discover": u"Az oldal Internet-c鉚m茅nek meg謾llap鉚t謾sa"} id = {"maintitle": "Kirim sebuah berkas", "submit": "Kirim", "sending": "Mengirim", "successtitle": "Berkas diterima", "received": "Berkas diterima!", "another": "Kirim berkas yang lain.", "errortitle": "Permasalahan", "problem": "Telah ditemukan sebuah kesalahan!", "retry": "Coba kembali.", "discover": "Kenali alamat IP dari halaman ini"} it = {"maintitle": u"Invia un file", "submit": u"Invia", "sending": u"Invio in corso", "successtitle": u"File ricevuto", "received": u"File ricevuto!", "another": u"Invia un altro file.", "errortitle": u"Errore", "problem": u"Si 貓 verificato un errore!", "retry": u"Riprova.", "discover": u"Scopri l鈥檌ndirizzo di questa pagina"} ja = {"maintitle": u"銉曘偂銈ゃ儷閫佷俊", "submit": u"閫佷俊", "sending": u"閫佷俊涓�", "successtitle": u"鍙椾俊瀹屼簡", "received": u"銉曘偂銈ゃ儷銈掑彈淇°仐銇俱仐銇燂紒", "another": u"浠栥伄銉曘偂銈ゃ儷銈掗€佷俊銇欍倠", "errortitle": u"鍟忛鐧虹敓", "problem": u"鍟忛銇岀櫤鐢熴仐銇俱仐銇燂紒", "retry": u"銉儓銉┿偆", "discover": u"銇撱伄銉氥兗銈搞伄銈儔銉偣銈掔⒑瑾嶃仚銈�"} ko = {"maintitle": u"韺岇澕氤措偞旮�", "submit": u"氤措偞旮�", "sending": u"氤措偞電� 欷�", "successtitle": u"韺岇澕鞚� 氚涭晞臁岇姷雼堧嫟", "received": u"韺岇澕鞚� 氚涭晞臁岇姷雼堧嫟!", "another": u"雼るジ韺岇澕氤措偞旮�", "errortitle": u"氍胳牅臧€ 氚滌儩頄堨姷雼堧嫟", "problem": u"氍胳牅臧€ 氚滌儩頄堨姷雼堧嫟!", "retry": u"雼れ嫓鞁滊弰", "discover": u"鞚� 韼橃澊歆€ 歟檢唽鞎岇晞氤搓赴"} nl = {"maintitle": "Verstuur een bestand", "submit": "Verstuur", "sending": "Bezig met versturen", "successtitle": "Bestand ontvangen", "received": "Bestand ontvangen!", "another": "Verstuur nog een bestand.", "errortitle": "Fout", "problem": "Er is een fout opgetreden!", "retry": "Nog eens.", "discover": "Vind het adres van deze pagina"} no = {"maintitle": u"Send en fil", "submit": u"Send", "sending": u"Sender", "successtitle": u"Fil mottatt", "received": u"Fil mottatt !", "another": u"Send en ny fil.", "errortitle": u"Feil", "problem": u"Det har skjedd en feil !", "retry": u"Send p氓 nytt.", "discover": u"Finn addressen til denne siden"} pl = {"maintitle": u"Wy艣lij plik", "submit": u"Wy艣lij", "sending": u"Wysy艂anie", "successtitle": u"Plik wys艂any", "received": u"Plik wys艂any!", "another": u"Wy艣lij kolejny plik.", "errortitle": u"Problem", "problem": u"Wyst膮pi艂 b艂膮d!", "retry": u"Spr貿buj ponownie.", "discover": u"Znajd藕 adres tej strony"} pt = {"maintitle": u"Enviar um ficheiro", "submit": u"Enviar", "sending": u"A enviar", "successtitle": u"Ficheiro recebido", "received": u"Ficheiro recebido !", "another": u"Enviar outro ficheiro.", "errortitle": u"Erro", "problem": u"Ocorreu um erro !", "retry": u"Tentar novamente.", "discover": u"Descobrir o endere莽o desta p謾gina"} pt_br = { "maintitle": u"Enviar um arquivo", "submit": u"Enviar", "sending": u"Enviando", "successtitle": u"Arquivo recebido", "received": u"Arquivo recebido!", "another": u"Enviar outro arquivo.", "errortitle": u"Erro", "problem": u"Ocorreu um erro!", "retry": u"Tentar novamente.", "discover": u"Descobrir o endere莽o desta p謾gina"} ro = {"maintitle": u"Trimite un fi艧ier", "submit": u"Trimite", "sending": u"Se trimite", "successtitle": u"Fi艧ier recep牛ionat", "received": u"Fi艧ier recep牛ionat !", "another": u"Trimite un alt fi艧ier.", "errortitle": u"Problem膬", "problem": u"A intervenit o problem膬 !", "retry": u"Re卯ncearc膬.", "discover": u"Descoper膬 adresa acestei pagini"} ru = {"maintitle": u"袨褌鋅褉邪脅懈褌褜褎邪泄謝", "submit": u"袨褌鋅褉邪脅懈褌褜", "sending": u"袨褌鋅褉邪脅謝褟褞", "successtitle": u"肖邪泄謝鋅芯謝褍褔械薪", "received": u"肖邪泄謝鋅芯謝褍褔械薪 !", "another": u"袨褌鋅褉邪脅懈褌褜寫褉褍諧芯泄褎邪泄謝.", "errortitle": u"袨褕懈斜瀉邪", "problem": u"袩褉芯懈蟹芯褕謝邪芯褕懈斜瀉邪 !", "retry": u"袩芯脅褌芯褉懈褌褜.", "discover": u"袩芯褋屑芯褌褉械褌褜邪寫褉械褋褝褌芯泄褋褌褉邪薪懈褑褘"} sk = {"maintitle": u"Po擰li s煤bor", "submit": u"Po擰li", "sending": u"Posielam", "successtitle": u"S煤bor prijat媒", "received": u"S煤bor prijat媒 !", "another": u"Posla鈕膹al擰鉚 s煤bor.", "errortitle": u"Chyba", "problem": u"Vyskytla sa chyba!", "retry": u"Sk煤si鈕 znova.", "discover": u"Zisti adresu tejto str謾nky"} sl = {"maintitle": u"Po擰lji datoteko", "submit": u"Po擰lji", "sending": u"Po擰iljam", "successtitle": u"Datoteka prejeta", "received": u"Datoteka prejeta !", "another": u"Po擰lji novo datoteko.", "errortitle": u"Napaka", "problem": u"Pri擰lo je do napake !", "retry": u"Poizkusi ponovno.", "discover": u"Poi擰膷i naslov na tej strani"} sr = {"maintitle": u"Po擰alji fajl", "submit": u"Po擰alji", "sending": u"艩aljem", "successtitle": u"Fajl primljen", "received": u"Fajl primljen !", "another": u"Po擰alji jo擰 jedan fajl.", "errortitle": u"Problem", "problem": u"Desio se problem !", "retry": u"Poku擰aj ponovo.", "discover": u"Otkrij adresu ove stranice"} sv = {"maintitle": u"Skicka en fil", "submit": u"Skicka", "sending": u"Skickar...", "successtitle": u"Fil mottagen", "received": u"Fil mottagen !", "another": u"Skicka en fil till.", "errortitle": u"Fel", "problem": u"Det har uppst氓tt ett fel !", "retry": u"F枚rs枚k igen.", "discover": u"Ta reda p氓 adressen till denna sida"} tr = {"maintitle": u"Dosya g枚nder", "submit": u"G枚nder", "sending": u"G枚nderiliyor...", "successtitle": u"G枚nderildi", "received": u"G枚nderildi", "another": u"Ba艧ka bir dosya g枚nder.", "errortitle": u"Problem.", "problem": u"Bir problem oldu !", "retry": u"Yeniden dene.", "discover": u"Bu sayfan謀n adresini bul"} zh_cn = { "maintitle": u"鍙戦€佹枃浠�", "submit": u"鍙戦€�", "sending": u"鍙戦€佷腑", "successtitle": u"鏂囦歡宸叉敹鍒�", "received": u"鏂囦歡宸叉敹鍒幫紒", "another": u"鍙戦€佸彠涓€涓枃浠躲€�", "errortitle": u"闂", "problem": u"鍑虹幇闂錛�", "retry": u"閲嶈瘯銆�", "discover": u"鏌ョ湅鏈〉闈㈢殑鍦板潃"} translations = {"ar": ar, "cs": cs, "da": da, "de": de, "el": el, "en": en, "es": es, "fi": fi, "fr": fr, "gl": gl, "hu": hu, "id": id, "it": it, "ja": ja, "ko": ko, "nl": nl, "no": no, "pl": pl, "pt": pt, "pt-br": pt_br, "ro": ro, "ru": ru, "sk": sk, "sl": sl, "sr": sr, "sv": sv, "tr": tr, "zh-cn": zh_cn} class DroopyFieldStorage(cgi.FieldStorage): """The file is created in the destination directory and its name is stored in the tmpfilename attribute. """ TMPPREFIX = 'tmpdroopy' def make_file(self, binary=None): fd, name = tempfile.mkstemp(dir=directory, prefix=self.TMPPREFIX) self.tmpfile = os.fdopen(fd, 'w+b') self.tmpfilename = name return self.tmpfile class HTTPUploadHandler(BaseHTTPServer.BaseHTTPRequestHandler): protocol_version = 'HTTP/1.0' form_field = 'upfile' divpicture = '<div class="box"><img src="/__droopy/picture"/></div>' def html(self, page): """ page can be "main", "success", or "error" returns an html page (in the appropriate language) as a string """ # -- Parse accept-language header if not self.headers.has_key("accept-language"): a = [] else: a = self.headers["accept-language"] a = a.split(',') a = [e.split(';q=') for e in a] a = [(lambda x: len(x)==1 and (1, x[0]) or (float(x[1]), x[0])) (e) for e in a] a.sort() a.reverse() a = [x[1] for x in a] # now a is an ordered list of preferred languages # -- Choose the appropriate translation dictionary (default is english) lang = "en" for l in a: if translations.has_key(l): lang = l break dico = copy.copy(translations[lang]) # -- Set message and picture dico["message"] = message if picture != None: dico["divpicture"] = self.divpicture else: dico["divpicture"] = "" # -- Possibly provide download links links = "" names = self.published_files() if names: for name in names: links += '<a href="/%s">%s</a><br/>' % ( urllib.quote(name.encode('utf-8')), name) links = '<div id="files">' + links + '</div>' dico["files"] = links # -- Add a link to discover the url if self.client_address[0] == "127.0.0.1": dico["port"] = self.server.server_port dico["linkurl"] = linkurltmpl % dico else: dico["linkurl"] = "" return templates[page] % dico def do_GET(self): name = self.path.lstrip('/') name = urllib.unquote(name) name = name.decode('utf-8') if picture != None and self.path == '/__droopy/picture': # send the picture self.send_file(picture) elif name in self.published_files(): localpath = os.path.join(directory, name) self.send_file(localpath) else: self.send_html(self.html("main")) def do_POST(self): # Do some browsers /really/ use multipart ? maybe Opera ? try: self.log_message("Started file transfer") # -- Set up environment for cgi.FieldStorage env = {} env['REQUEST_METHOD'] = self.command if self.headers.typeheader is None: env['CONTENT_TYPE'] = self.headers.type else: env['CONTENT_TYPE'] = self.headers.typeheader # -- Save file (numbered to avoid overwriting, ex: foo-3.png) form = DroopyFieldStorage(fp = self.rfile, environ = env); fileitem = form[self.form_field] filename = self.basename(fileitem.filename).decode('utf-8') if filename == "": self.send_response(303) self.send_header('Location', '/') self.end_headers() return localpath = os.path.join(directory, filename).encode('utf-8') root, ext = os.path.splitext(localpath) i = 1 # race condition, but hey... while (os.path.exists(localpath)): localpath = "%s-%d%s" % (root, i, ext) i = i+1 if hasattr(fileitem, 'tmpfile'): # DroopyFieldStorage.make_file() has been called fileitem.tmpfile.close() shutil.move(fileitem.tmpfilename, localpath) else: # no temporary file, self.file is a StringIO() # see cgi.FieldStorage.read_lines() fout = file(localpath, 'wb') shutil.copyfileobj(fileitem.file, fout) fout.close() self.log_message("Received: %s", os.path.basename(localpath)) # -- Reply if publish_files: # The file list gives a feedback for the upload # success self.send_response(301) self.send_header("Location", "/") self.end_headers() else: self.send_html(self.html("success")) except Exception, e: self.log_message(repr(e)) self.send_html(self.html("error")) def send_html(self, htmlstr): self.send_response(200) self.send_header('Content-type','text/html; charset=utf-8') self.end_headers() self.wfile.write(htmlstr.encode('utf-8')) def send_file(self, localpath): f = open(localpath, 'rb') self.send_response(200) self.send_header('Content-type', mimetypes.guess_type(localpath)[0]) self.send_header('Content-length', os.fstat(f.fileno())[6]) self.end_headers() shutil.copyfileobj(f, self.wfile) def basename(self, path): """Extract the file base name (some browsers send the full file path). """ for mod in posixpath, macpath, ntpath: path = mod.basename(path) return path def published_files(self): """Returns the list of files that should appear as download links. The returned filenames are unicode strings. """ if publish_files: # os.listdir() returns a list of unicode strings when the # directory is passed as an unicode string itself. names = [name for name in os.listdir(unicode(directory)) if os.path.isfile(os.path.join(directory, name)) and not name.startswith(DroopyFieldStorage.TMPPREFIX)] names.sort() else: names = [] return names def handle(self): try: BaseHTTPServer.BaseHTTPRequestHandler.handle(self) except socket.error, e: self.log_message(str(e)) raise Abort() class Abort(Exception): pass class ThreadedHTTPServer(SocketServer.ThreadingMixIn, BaseHTTPServer.HTTPServer): def handle_error(self, request, client_address): # Override SocketServer.handle_error exctype = sys.exc_info()[0] if not exctype is Abort: BaseHTTPServer.HTTPServer.handle_error(self,request,client_address) # -- Options def configfile(): appname = 'droopy' # os.name is 'posix', 'nt', 'os2', 'mac', 'ce' or 'riscos' if os.name == 'posix': filename = "%s/.%s" % (os.environ["HOME"], appname) elif os.name == 'mac': filename = ("%s/Library/Application Support/%s" % (os.environ["HOME"], appname)) elif os.name == 'nt': filename = ("%s\%s" % (os.environ["APPDATA"], appname)) else: filename = None return filename def save_options(): opt = [] if message: opt.append('--message=%s' % message.replace('\n', '\\n')) if picture: opt.append('--picture=%s' % picture) if directory: opt.append('--directory=%s' % directory) if publish_files: opt.append('--dl') if port: opt.append('%d' % port) f = open(configfile(), 'w') f.write('\n'.join(opt).encode('utf8')) f.close() def load_options(): try: f = open(configfile()) cmd = [line.strip().decode('utf8').replace('\\n', '\n') for line in f.readlines()] parse_args(cmd) f.close() return True except IOError, e: return False def parse_args(cmd=None): """Parse command-line arguments. Parse sys.argv[1:] if no argument is passed. """ global picture, message, port, directory, must_save_options, publish_files if cmd == None: cmd = sys.argv[1:] lang, encoding = locale.getdefaultlocale() if encoding != None: cmd = [a.decode(encoding) for a in cmd] opts, args = None, None try: opts, args = getopt.gnu_getopt(cmd, "p:m:d:h", ["picture=","message=", "directory=", "help", "save-config","delete-config", "dl"]) except Exception, e: print e sys.exit(1) for o,a in opts: if o in ["-p", "--picture"] : picture = os.path.expanduser(a) elif o in ["-m", "--message"] : message = '<div id="message" class="bubble">%s </div>' % a elif o in ['-d', '--directory']: directory = a elif o in ['--save-config']: must_save_options = True elif o in ['--delete-config']: try: filename = configfile() os.remove(filename) print 'Deleted ' + filename except Exception, e: print e sys.exit(0) elif o in ['--dl']: publish_files = True elif o in ['-h', '--help']: print USAGE sys.exit(0) # port number try: if args[0:]: port = int(args[0]) except ValueError: print args[0], "is not a valid port number" sys.exit(1) # -- def run(): """Run the webserver.""" socket.setdefaulttimeout(3*60) server_address = ('', port) httpd = ThreadedHTTPServer(server_address, HTTPUploadHandler) httpd.serve_forever() if __name__ == '__main__': print LOGO config_found = load_options() parse_args() if config_found: print 'Configuration found in %s' % configfile() else: print "No configuration file found." if must_save_options: save_options() print "Options saved in %s" % configfile() print "Files will be uploaded to %s" % directory try: print print "HTTP server running... Check it out at http://localhost:%d"%port run() except KeyboardInterrupt: print '^C received, shutting down server' # some threads may run until they terminate