-
Notifications
You must be signed in to change notification settings - Fork 2
/
new-bot.py
40 lines (38 loc) · 1.36 KB
/
new-bot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#!/usr/bin/env python
import os
print "Create a new Twitter account: https://twitter.com/signup"
print "What is the Twitter username for your bot?"
BOT_NAME = raw_input()
with open('bots.list', 'a') as bot_list:
bot_list.write(BOT_NAME + "\n")
os.mkdir(BOT_NAME)
CFG_FILE = BOT_NAME + os.sep + "omni.cfg"
CFG = "[" + BOT_NAME + "]\n"
CFG += "handle=" + BOT_NAME + "\n"
print "Create an app for this username at https://dev.twitter.com/apps/new"
OAUTH_FILE = BOT_NAME + "_oauth"
CFG += "oauth=" + OAUTH_FILE + "\n"
print "What is your Consumer Key?"
CONSUMER_KEY = raw_input()
CFG += "consumer_key=" + CONSUMER_KEY + "\n"
print "What is your Consumer Secret?"
CONSUMER_SECRET = raw_input()
CFG += "consumer_secret=" + CONSUMER_SECRET + "\n"
print "What is your OAuth Access Token?"
ACCESS_TOKEN = raw_input()
print "What is your OAuth Access Token Secret?"
ACCESS_SECRET = raw_input()
with open(OAUTH_FILE, 'w') as o:
o.write(ACCESS_TOKEN + "\n")
o.write(ACCESS_SECRET + "\n")
print "OK, now for the content of bot"
print "What is the name of the corpus file for your bot?"
FILE_NAME = raw_input()
CFG += "corpus=" + FILE_NAME + "\n"
print "What method should we use to generate text from the corpus?"
print "Choose between markov, isopsephy, or random"
METHOD = raw_input()
CFG += "tweet_method=" + METHOD + "\n"
with open(CFG_FILE, 'w') as cfg:
cfg.write(CFG)
print "Done!"