Various fixes

- more pep8
- add some forgotten deps in setup.py
This commit is contained in:
jvoisin 2015-07-25 10:37:05 +02:00
parent d50601f5bd
commit ac97ddf7d0
6 changed files with 22 additions and 19 deletions

View File

@ -40,6 +40,7 @@ if get_platform() == 'Darwin':
else: else:
osx_resources_dir = None osx_resources_dir = None
def get_onionshare_dir(): def get_onionshare_dir():
if get_platform() == 'Darwin': if get_platform() == 'Darwin':
onionshare_dir = os.path.dirname(__file__) onionshare_dir = os.path.dirname(__file__)
@ -47,6 +48,7 @@ def get_onionshare_dir():
onionshare_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) onionshare_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
return onionshare_dir return onionshare_dir
def get_html_path(filename): def get_html_path(filename):
p = platform.system() p = platform.system()
if p == 'Darwin': if p == 'Darwin':

View File

@ -17,7 +17,7 @@ GNU General Public License for more details.
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import os, sys, subprocess, time, argparse, inspect, shutil, socket, threading, urllib2, httplib, tempfile import os, sys, subprocess, time, argparse, shutil, socket, threading, urllib2, httplib, tempfile
import socks import socks
from stem.control import Controller from stem.control import Controller
@ -52,6 +52,7 @@ class OnionShare(object):
self.port = None self.port = None
self.controller = None self.controller = None
self.hidserv_dir = None self.hidserv_dir = None
self.onion_host = None
# debug mode # debug mode
if debug: if debug:
@ -112,14 +113,14 @@ class OnionShare(object):
args = ['/usr/bin/sudo', '--', '/usr/bin/onionshare'] args = ['/usr/bin/sudo', '--', '/usr/bin/onionshare']
print "Executing: {0:s}".format(args+[str(self.port)]) print "Executing: {0:s}".format(args+[str(self.port)])
p = subprocess.Popen(args+[str(self.port)], stderr=subprocess.PIPE, stdout=subprocess.PIPE) p = subprocess.Popen(args+[str(self.port)], stderr=subprocess.PIPE, stdout=subprocess.PIPE)
stdout = p.stdout.read(22) # .onion URLs are 22 chars long stdout = p.stdout.read(22) # .onion URLs are 22 chars long
if stdout: if stdout:
self.onion_host = stdout self.onion_host = stdout
print 'Got onion_host: {0:s}'.format(self.onion_host) print 'Got onion_host: {0:s}'.format(self.onion_host)
else: else:
if p.poll() == -1: if p.poll() == -1:
raise TailsError(o.stderr.read()) raise TailsError(sys.stderr.read())
else: else:
raise TailsError(strings._("error_tails_unknown_root")) raise TailsError(strings._("error_tails_unknown_root"))
@ -227,7 +228,7 @@ class OnionShare(object):
except urllib2.HTTPError: # Tails error except urllib2.HTTPError: # Tails error
sys.stdout.write('{0:s}\n'.format(strings._('wait_for_hs_nope'))) sys.stdout.write('{0:s}\n'.format(strings._('wait_for_hs_nope')))
sys.stdout.flush() sys.stdout.flush()
except httplib.BadStatusLine: # Tails (with bridge) error except httplib.BadStatusLine: # Tails (with bridge) error
sys.stdout.write('{0:s}\n'.format(strings._('wait_for_hs_nope'))) sys.stdout.write('{0:s}\n'.format(strings._('wait_for_hs_nope')))
sys.stdout.flush() sys.stdout.flush()
except KeyboardInterrupt: except KeyboardInterrupt:

View File

@ -198,8 +198,8 @@ class socksocket(socket.socket):
default_proxy = None default_proxy = None
def __init__(self, family=socket.AF_INET, type=socket.SOCK_STREAM, proto=0, _sock=None): def __init__(self, family=socket.AF_INET, socket_type=socket.SOCK_STREAM, proto=0, _sock=None):
_orig_socket.__init__(self, family, type, proto, _sock) _orig_socket.__init__(self, family, socket_type, proto, _sock)
if self.default_proxy: if self.default_proxy:
self.proxy = self.default_proxy self.proxy = self.default_proxy

View File

@ -17,7 +17,7 @@ GNU General Public License for more details.
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import json, locale, sys, os, inspect import json, locale, sys, os
import helpers import helpers
strings = {} strings = {}
@ -36,22 +36,22 @@ def load_strings(default="en"):
locale_dir = os.path.join(os.path.dirname(helpers.get_onionshare_dir()), 'locale') locale_dir = os.path.join(os.path.dirname(helpers.get_onionshare_dir()), 'locale')
# load all translations # load all translations
translated = {} translations = {}
for filename in os.listdir(locale_dir): for filename in os.listdir(locale_dir):
abs_filename = os.path.join(locale_dir, filename) abs_filename = os.path.join(locale_dir, filename)
lang, ext = os.path.splitext(filename) lang, ext = os.path.splitext(filename)
if abs_filename.endswith('.json'): if abs_filename.endswith('.json'):
translated[lang] = json.loads(open(abs_filename).read()) translations[lang] = json.loads(open(abs_filename).read())
strings = translated[default] strings = translations[default]
lc, enc = locale.getdefaultlocale() lc, enc = locale.getdefaultlocale()
if lc: if lc:
lang = lc[:2] lang = lc[:2]
if lang in translated: if lang in translations:
# if a string doesn't exist, fallback to English # if a string doesn't exist, fallback to English
for key in translated[default]: for key in translations[default]:
if key in translated[lang]: if key in translations[lang]:
strings[key] = translated[lang][key] strings[key] = translations[lang][key]
def translated(k, gui=False): def translated(k, gui=False):

View File

@ -17,7 +17,7 @@ GNU General Public License for more details.
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import Queue, mimetypes, platform, os, sys, zipfile, urllib2 import Queue, mimetypes, platform, os, sys, urllib2
from flask import Flask, Response, request, render_template_string, abort from flask import Flask, Response, request, render_template_string, abort
import strings, helpers import strings, helpers
@ -70,10 +70,10 @@ REQUEST_CANCELED = 4
q = Queue.Queue() q = Queue.Queue()
def add_request(type, path, data=None): def add_request(request_type, path, data=None):
global q global q
q.put({ q.put({
'type': type, 'type': request_type,
'path': path, 'path': path,
'data': data 'data': data
}) })
@ -146,7 +146,7 @@ def download(slug_candidate):
basename = os.path.basename(zip_filename) basename = os.path.basename(zip_filename)
def generate(): def generate():
chunk_size = 102400 # 100kb chunk_size = 102400 # 100kb
fp = open(zip_filename, 'rb') fp = open(zip_filename, 'rb')
done = False done = False

View File

@ -123,5 +123,5 @@ elif system == 'Darwin':
'PyQt4.QtSvg', 'PyQt4.QtXmlPatterns'] 'PyQt4.QtSvg', 'PyQt4.QtXmlPatterns']
} }
}, },
setup_requires=['py2app'], setup_requires=['py2app', 'flask', 'stem'],
) )