manage.py

#!/usr/bin/env python3

from models.conversation import Conversation
from models.user import User
from models.command import Command
from models.message import Message
from models.hook import Hook

from models.recommendation import Recommendation
from models.lastuser import LastUser

from hangoutsbot import HangoutsBot

from manager import Manager
from colors import red

import settings

import sys
import hangups
import asyncio
import code

manager = Manager()

def create_the_tables(tables):
    for table in tables:
        if table.table_exists():
            print("Table already exists for {}".format(table))
        else:
            table.create_table()
            print("Created table for {}".format(table))


@manager.command
def create_tables():
    """Create the tables for the models in the database"""
    tables = [Conversation, User, Command, Message, Conversation.members.get_through_model(), Hook]
    create_the_tables(tables)


@manager.command
def create_extra_tables():
    """Create the extra tables for the models in the database"""
    tables = [Recommendation, LastUser]
    create_the_tables(tables)


@manager.command
def run():
    """Run HangoutsBot"""
    bot = HangoutsBot()
    bot.run()


@manager.command
def shell():
    """Open python shell with context"""
    code.interact(local=globals())


@manager.command
def get_bot_id():
    """Get ID of Bot Account to put in secret.py"""
    client = hangups.client.Client(hangups.auth.get_auth_stdin(settings.COOKIES_FILE_PATH))
    client.on_connect.add_observer(lambda: asyncio.async(get_id(client)))
    loop = asyncio.get_event_loop()
    loop.run_until_complete(client.connect())


@asyncio.coroutine
def get_id(client):
    request = hangups.hangouts_pb2.GetSelfInfoRequest(
        request_header=client.get_request_header(),
    )
    try:
        response = yield from client.get_self_info(request)
        print(response.self_entity.id.gaia_id)
    finally:
        yield from client.disconnect()


if __name__ == '__main__':
    if sys.version_info < (3, 4):
        print(red("python3.4 is required to use hangoutsbot"))
        sys.exit(1)
    manager.main()