matt/markov-bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added command-line option

Browse Source
This commit is contained in:
Vanessa McHale
2017-03-13 00:53:46 -05:00
parent 6abc17df97
commit 1fdfa3c9b9
1 changed files with 30 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
markov.py
View File

@@ -2,26 +2,36 @@ import markovify
import twitter import twitter
import itertools import itertools
import argparse import argparse
import random
## Command line parser # Command line parser
parser = argparse.ArgumentParser(description='Process some integers.') parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('filepath', metavar='CREDENTIALS', type=str, help='an integer for the accumulator') parser.add_argument('filepath', metavar='CREDENTIALS', type=str, help='an integer for the accumulator')
parser.add_argument('--text', dest='corpus', metavar='CORPUS', type=str, help='path to the text you wish to mimic') parser.add_argument('--text', dest='corpus', metavar='CORPUS', type=str, help='path to the text you wish to mimic') # filepath should be a flag; corpus an argument!
parser.add_argument('-t', dest='test', action="store_true", default=False, help='Test text generation without tweeting.') # distance on markov chains?? parser.add_argument('--coke-binge', dest='coke_binge', action="store_true", default=False, help='path to the text you wish to mimic') # instead of tweeting once, tweet a random number of times.
parser.add_argument('-t', dest='test', action="store_true", default=False, help='Test text generation without tweeting.') # distance on markov chains?? distance on syntax trees? # default to test or default to tweet??
args = parser.parse_args() args = parser.parse_args()
## Function to help process credentials file
# Function to help process credentials file
def drop_tag(str): def drop_tag(str):
parsed = itertools.dropwhile(lambda x: x!= ':', str) parsed = itertools.dropwhile(lambda x: x!= ':', str)
str2 = "".join(parsed) str2 = "".join(parsed)
parsed2 = itertools.dropwhile(lambda x: x == ' ' or x == ':', str2) parsed2 = itertools.dropwhile(lambda x: x == ' ' or x == ':', str2)
return "".join(parsed2) return "".join(parsed2)
## Open file with credentials
# Set the
if args.coke_binge:
coke_binge_num = random.randint(1, 6)
else:
coke_binge_num = 1
# Open file with credentials
with open(args.filepath) as f: with open(args.filepath) as f:
content = f.readlines() content = f.readlines()
## Process api keys etc. # Process api keys etc.
content = [drop_tag(x.strip()) for x in content] content = [drop_tag(x.strip()) for x in content]
# Get raw text of markov chain as string. # Get raw text of markov chain as string.
@@ -39,14 +49,16 @@ api = twitter.Api(consumer_key=content[0],
# Generate three randomly-generated sentences of no more than 140 characters and tweet them. # Generate three randomly-generated sentences of no more than 140 characters and tweet them.
selected = False selected = False
while selected == False: for i in range(0, coke_binge_num):
while selected is False:
str_potentially = text_model.make_short_sentence(140) str_potentially = text_model.make_short_sentence(140)
if not("http" in str_potentially): if not("http" in str_potentially):
if not(args.test): if not(args.test):
#tweet the generated text # tweet the generated text
status = api.PostUpdate(str_potentially) status = api.PostUpdate(str_potentially)
print(status.text) # verify it worked print(status.text) # verify it worked
else: else:
#test mode; just display it # test mode; just display it
print(str_potentially) print(str_potentially)
selected = True selected = True
selected = False