Optimise and minor bug fix random string generation

pull/66/head
David Trail 8 years ago
parent 65fbf90005
commit 92028f21d7

@ -7,9 +7,9 @@ import argparse
import json import json
import yaml import yaml
import praw import praw
import random
from re import sub from re import sub
from random import shuffle, randint
from datetime import datetime, timedelta from datetime import datetime, timedelta
from praw.errors import (InvalidUser, InvalidUserPass, RateLimitExceeded, from praw.errors import (InvalidUser, InvalidUserPass, RateLimitExceeded,
HTTPException, OAuthAppRequired) HTTPException, OAuthAppRequired)
@ -19,25 +19,6 @@ logging.basicConfig(stream=sys.stdout)
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
log.setLevel(level=logging.DEBUG) log.setLevel(level=logging.DEBUG)
try:
from loremipsum import get_sentence # This only works on Python 2
except ImportError:
def get_sentence():
return '''I have been Shreddited for privacy!'''
os_wordlist = '/usr/share/dict/words'
if os.name == 'posix' and os.path.isfile(os_wordlist):
# Generate a random string of words from our system's dictionary
fh = open(os_wordlist)
words = fh.read().splitlines()
fh.close()
shuffle(words)
def get_sentence():
return ' '.join(words[:randint(50, 150)])
assert get_sentence
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
'-c', '-c',
@ -63,7 +44,7 @@ if save_directory:
r.config.store_json_result = True r.config.store_json_result = True
if config.get('verbose', True): if config.get('verbose', True):
log_level = config.get('debug', 'WARNING') # Default to WARNING only log_level = config.get('debug', 'DEBUG')
log.setLevel(level=getattr(logging, log_level)) log.setLevel(level=getattr(logging, log_level))
try: try:
@ -98,6 +79,24 @@ if config.get('whitelist'):
) )
def get_sentence():
return '''I have been Shreddited for privacy!'''
try:
# Provide a method that works on windows
from loremipsum import get_sentence
except ImportError:
# Module unavailable, use the default phrase
pass
os_wordlist = '/usr/share/dict/words'
if os.name == 'posix' and os.path.isfile(os_wordlist):
# Generate a random string of words from our system's dictionary
fh = open(os_wordlist)
words = fh.read().splitlines()
fh.close()
def get_sentence():
return ' '.join(random.sample(words, random.randint(50,75)))
def get_things(after=None): def get_things(after=None):
limit = None limit = None
item = config.get('item', 'comments') item = config.get('item', 'comments')

Loading…
Cancel
Save