nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add command-line parameters to pastoob

Browse source 
And better defaults.
closes #920
This commit is contained in:
Laurent Bachelier 2013-01-25 17:25:25 +01:00
commit cf5b6a95f5
1 changed files with 42 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

50
weboob/applications/pastoob/pastoob.py
View file

@ -24,6 +24,7 @@ import os
import sys import sys
import codecs import codecs
import locale import locale
import re
from random import choice from random import choice
from weboob.capabilities.paste import ICapPaste, PasteNotFound from weboob.capabilities.paste import ICapPaste, PasteNotFound
@ -57,10 +58,10 @@ class Pastoob(ReplApplication):
try: try:
paste = self.get_object(_id, 'get_paste', ['contents']) paste = self.get_object(_id, 'get_paste', ['contents'])
except PasteNotFound: except PasteNotFound:
print >>sys.stderr, 'Paste not found: %s' % _id print >>sys.stderr, 'Paste not found: %s' % _id
return 3 return 3
if not paste: if not paste:
print >>sys.stderr, 'Unable to handle paste: %s' % _id print >>sys.stderr, 'Unable to handle paste: %s' % _id
return 1 return 1
output = codecs.getwriter(sys.stdout.encoding or locale.getpreferredencoding())(sys.stdout) output = codecs.getwriter(sys.stdout.encoding or locale.getpreferredencoding())(sys.stdout)
output.write(paste.contents) output.write(paste.contents)
@ -91,7 +92,7 @@ class Pastoob(ReplApplication):
return 1 return 1
# get and sort the backends able to satisfy our requirements # get and sort the backends able to satisfy our requirements
params = self._get_params() params = self.get_params()
backends = {} backends = {}
for backend in self.weboob.iter_backends(): for backend in self.weboob.iter_backends():
score = backend.can_post(contents, **params) score = backend.can_post(contents, **params)
@ -105,11 +106,44 @@ class Pastoob(ReplApplication):
return 1 return 1
p = backend.new_paste(_id=None) p = backend.new_paste(_id=None)
p.public = params.get('public') p.public = params['public']
p.title = os.path.basename(filename) if self.options.title is not None:
p.title = self.options.title
else:
p.title = os.path.basename(filename)
p.contents = contents p.contents = contents
backend.post_paste(p, max_age=params.get('max_age')) backend.post_paste(p, max_age=params['max_age'])
print 'Successfuly posted paste: %s' % p.page_url print 'Successfuly posted paste: %s' % p.page_url
def _get_params(self): def get_params(self):
return {'public': True, 'max_age': 3600*24*3} return {'public': self.options.public,
'max_age': self.str_to_duration(self.options.max_age),
'title': self.options.title}
def str_to_duration(self, s):
parts = re.findall(r'(\d*(?:\.\d+)?)\s*([A-z]+)', s)
argsmap = {'Y|y|year|years|yr|yrs': 365.25 * 24 * 3600,
'M|o|month|months': 30.5 * 24 * 3600,
'W|w|week|weeks': 7 * 24 * 3600,
'D|d|day|days': 24 * 3600,
'H|h|hours|hour|hr|hrs': 3600,
'm|i|minute|minutes|min|mins': 60,
'S|s|second|seconds|sec|secs': 1}
seconds = 0
for number, unit in parts:
for rx, secs in argsmap.iteritems():
if re.match('^(%s)$' % rx, unit):
seconds += float(number) * float(secs)
return int(seconds)
def add_application_options(self, group):
group.add_option('-p', '--public', action='store_true',
help='Make paste public.')
group.add_option('-t', '--title', action='store',
help='Paste title',
type='string')
group.add_option('-m', '--max-age', action='store',
help='Maximum age (duration), default "1 month"',
type='string', default='1 month')