from sqlalchemy import (Boolean, Column, Enum, BigInteger, Integer, String, 

    ForeignKey, DateTime, create_engine)

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy.orm import relationship

import anyjson

import datetime

import email.utils

Base = declarative_base()

APPLICATION_NAME = "IRI_TWITTER" 

CONSUMER_KEY = "54ThDZhpEjokcMgHJOMnQA"

CONSUMER_SECRET = "wUoL9UL2T87tfc97R0Dff2EaqRzpJ5XGdmaN2XK3udA"

ACCESS_TOKEN_KEY = None

ACCESS_TOKEN_SECRET = None

#ACCESS_TOKEN_KEY= "47312923-LiNTtz0I18YXMVIrFeTuhmH7bOvYsK6p3Ln2Dc"

#ACCESS_TOKEN_SECRET = "r3LoXVcjImNAElUpWqTu2SG2xCdWFHkva7xeQoncA"

def adapt_date(date_str):

    ts = email.utils.parsedate_tz(date_str) #@UndefinedVariable

    return datetime.datetime(*ts[0:7])

def adapt_json(obj):

    if obj is None:

        return None

    else:

        return anyjson.serialize(obj)

    __tablename__ = "tweet_process_event"

    id = Column(Integer, primary_key=True, autoincrement=True)

    args = Column(String)

    __tablename__ = "tweet_entity_type"

    id = Column(Integer, primary_key=True, autoincrement=True)

    label = Column(String)

    __tablename__ = "tweet_entity"

    id = Column(Integer, primary_key=True)

    tweet_id = Column(BigInteger, ForeignKey('tweet_tweet.id'))

    type = Column(String)

    entity_type_id = Column(Integer, ForeignKey('tweet_entity_type.id'), nullable=False)

    entity_type = relationship("EntityType", backref="entities")

    indice_start = Column(Integer)

    indice_end = Column(Integer)

    source = Column(String)

    __mapper_args__ = {'polymorphic_on': type, 'polymorphic_identity': 'entity_entity', 'with_polymorphic':'*'}

    __tablename__ = 'tweet_tweet_source'

    id = Column(Integer, primary_key=True, autoincrement=True)

    original_json = Column(String)

    TWEET_STATUS = {

        'OK' : 1,

        'ERROR' : 2,

        'NOT_TWEET': 3,

    }

    __tablename__ = 'tweet_tweet_log'

    id = Column(Integer, primary_key=True, autoincrement=True)

    tweet_source_id = Column(Integer, ForeignKey('tweet_tweet_source.id'))

    tweet_source = relationship("TweetSource", backref="logs")

    status = Column(Integer)

    error = Column(String)

    error_stack = Column(String)

    __tablename__ = 'tweet_tweet'

    id = Column(BigInteger, primary_key=True, autoincrement=False)

    id_str = Column(String)

    contributors = Column(String)

    coordinates = Column(String) 

    created_at = Column(DateTime)

    favorited = Column(Boolean)

    geo = Column(String)

    in_reply_to_screen_name = Column(String)

    in_reply_to_status_id = Column(BigInteger) 

    in_reply_to_status_id_str = Column(String)

    in_reply_to_user_id = Column(BigInteger)

    in_reply_to_user_id_str = Column(String)

    place = Column(String)

    retweet_count = Column(String)

    retweeted = Column(Boolean)

    source = Column(String)

    text = Column(String)

    truncated = Column(Boolean)

    user_id = Column(Integer, ForeignKey('tweet_user.id'))

    user = relationship("User", backref="tweets")

    tweet_source_id = Column(Integer, ForeignKey('tweet_tweet_source.id'))

    tweet_source = relationship("TweetSource", backref="tweet")

    entity_list = relationship(Entity, backref='tweet')

    __tablename__ = "tweet_user_message"

    id = Column(Integer, primary_key=True)

    user_id = Column(Integer, ForeignKey('tweet_user.id'))

    user = relationship("User", backref="messages")

    message_id = Column(Integer, ForeignKey('tweet_message.id'))

    __tablename__ = "tweet_message"

    id = Column(Integer, primary_key=True)

    text = Column(String)

    users = relationship(UserMessage, backref='message')

    __tablename__ = "tweet_user"

    id = Column(BigInteger, primary_key=True, autoincrement=False)

    id_str = Column(String)

    contributors_enabled = Column(Boolean)

    created_at = Column(DateTime)

    description = Column(String)

    favourites_count = Column(Integer)

    follow_request_sent = Column(Boolean)

    followers_count = Column(Integer)

    following = Column(String)

    friends_count = Column(Integer)

    geo_enabled = Column(Boolean)

    is_translator = Column(Boolean)

    lang = Column(String)

    listed_count = Column(Integer)

    location = Column(String)

    name = Column(String)

    notifications = Column(String)

    profile_background_color = Column(String)

    profile_background_image_url = Column(String)

    profile_background_tile = Column(Boolean)

    profile_image_url = Column(String)

    profile_link_color = Column(String)

    profile_sidebar_border_color = Column(String)

    profile_sidebar_fill_color = Column(String)

    profile_text_color = Column(String)

    profile_use_background_image = Column(Boolean)

    protected = Column(Boolean)

    screen_name = Column(String, index=True)

    show_all_inline_media = Column(Boolean)

    statuses_count = Column(Integer)

    time_zone = Column(String)

    url = Column(String)

    utc_offset = Column(Integer)

    verified = Column(Boolean)

    __tablename__ = "tweet_hashtag"

    id = Column(Integer, primary_key=True)

    text = Column(String, unique=True, index=True)

    __tablename__ = "tweet_url"

    id = Column(Integer, primary_key=True)

    url = Column(String, unique=True)

    expanded_url = Column(String)

    __tablename__ = "tweet_media_type"

    id = Column(Integer, primary_key=True, autoincrement=True)

    label = Column(String, unique=True, index=True)

    __tablename__ = "tweet_media"

    id = Column(BigInteger, primary_key=True, autoincrement=False)

    id_str = Column(String, unique=True)

    media_url = Column(String, unique=True)

    media_url_https = Column(String, unique=True)

    url = Column(String)

    display_url = Column(String)

    expanded_url = Column(String)

    sizes = Column(String)

    type_id = Column(Integer, ForeignKey("tweet_media_type.id"))

    type = relationship(MediaType, primaryjoin=type_id == MediaType.id)

class EntityHashtag(Entity):

    __tablename__ = "tweet_entity_hashtag"

    __mapper_args__ = {'polymorphic_identity': 'entity_hashtag'}

    id = Column(Integer, ForeignKey('tweet_entity.id'), primary_key=True)

    hashtag_id = Column(Integer, ForeignKey("tweet_hashtag.id"))

    hashtag = relationship(Hashtag, primaryjoin=hashtag_id == Hashtag.id)

class EntityUrl(Entity):

    __tablename__ = "tweet_entity_url"

    __mapper_args__ = {'polymorphic_identity': 'entity_url'}

    id = Column(Integer, ForeignKey('tweet_entity.id'), primary_key=True)

    url_id = Column(Integer, ForeignKey("tweet_url.id"))

    url = relationship(Url, primaryjoin=url_id == Url.id)

class EntityUser(Entity):

    __tablename__ = "tweet_entity_user"

    __mapper_args__ = {'polymorphic_identity': 'entity_user'}

    id = Column(Integer, ForeignKey('tweet_entity.id'), primary_key=True)

    user_id = Column(BigInteger, ForeignKey('tweet_user.id'))

    user = relationship(User, primaryjoin=(user_id == User.id))

class EntityMedia(Entity):

    __tablename__ = "tweet_entity_media"

    __mapper_args__ = {'polymorphic_identity': 'entity_media'}

    id = Column(Integer, ForeignKey('tweet_entity.id'), primary_key=True)

    media_id = Column(BigInteger, ForeignKey('tweet_media.id'))

    media = relationship(Media, primaryjoin=(media_id == Media.id))

def setup_database(*args, **kwargs):

    create_all = True

    if "create_all" in kwargs:

        create_all = kwargs["create_all"]

        del(kwargs["create_all"])

    engine = create_engine(*args, **kwargs)

    metadata = Base.metadata

    if create_all:

        metadata.create_all(engine)

    return (engine, metadata)