def main():
try:
print 'connecting to database'
db = database.create_database()
c = db.cursor()
print 'connected'
parser = argparse.ArgumentParser('MathChallenge Message Sender')
parser.add_argument('--to', choices=['all', 'user'], help='Who are you sending this to?')
parser.add_argument('--user', nargs=1, default=None)
parser.add_argument('--message', help='message to be send', nargs=1)
parser.add_argument('--file', help='file to read to send message', type=argparse.FileType('r'))
args = parser.parse_args()
if args.message and args.file:
print 'Give one of file or message not both'
return
message = None
if args.file:
message = args.file.read()
args.file.close()
elif args.message:
message = args.message[0]
if message is None:
print 'No message to send'
return
if args.to == 'user' and args.user:
print 'finding ', args.user
c.execute('SELECT username, mxit_userid FROM users WHERE mxit_userid = %s or username = %s', (args.user[0], args.user[0]))
elif args.to == 'all':
c.execute('SELECT username, mxit_userid FROM users')
else:
print 'No users to select'
return
print 'connecting to mxit API'
api=MxitAPI(config.client_id, config.secret_id, 'mathchallenge')
api.auth(('message/send',))
print 'connected'
for i, (username, userid) in enumerate(c):
if username == 'Yasen' or i < 7324:
continue
r = api.send_message(userid, message.replace('<user>', username))
print i
print 'OK' if r.ok else 'FAILURE', username
if not r.ok:
print r.text
break
finally:
c.close()
db.close()
def getFsMigrations():
return set(filename for filename in os.listdir(MIGRATION_DIR)
if filename.endswith('.sql'))
if __name__ == '__main__':
parser = argparse.ArgumentParser('Reader migrator')
# if this argument isn't present we're only doing a dry run
parser.add_argument('--run', help='Do the real deal and migrate',
action='store_const', const=True)
args = parser.parse_args()
dryRun = not args.run
db = database.create_database()
createMigrationsTable(db)
dbMigrations = getDbMigrations(db)
fsMigrations = getFsMigrations()
incompleteMigrations = list(fsMigrations - dbMigrations)
incompleteMigrations.sort()
c = db.cursor()
for i, migration in enumerate(incompleteMigrations):
print 'Running migration %d: %s' % (i, migration)
with open(MIGRATION_DIR+migration) as f:
c.execute(f.read())
c.execute('INSERT INTO migrations (name) VALUES (%s)',
