#!/usr/bin/env python

from db import db as DB
from pprint import pprint

class MigrateDB(DB):

    def __init__(self, *args, **kwargs):
        super(MigrateDB, self).__init__(*args, **kwargs)

    def get_table(self, name):
        self.execute('SELECT * FROM %s' % name)
        fields = [d[0] for d in self.cursor.description]
        rows = self.cursor.fetchall()
        for row in rows:
            yield dict(zip(fields, row))

    def get_radio_chat_table(self):
        return self.get_table('radio_chat')

    def get_search_log_table(self):
        return self.get_table('search_log')

    def get_sj_chat_table(self):
        return self.get_table('sj_chat')

    def get_sj_likes_table(self):
        return self.get_table('sj_likes')

    def get_sj_likes_tmp_table(self):
        return self.get_table('sj_likes_tmp')

    def get_sj_room_table(self):
        return self.get_table('sj_room')

    def get_sj_search_log_table(self):
        return self.get_table('sj_search_log')

    def get_sj_session_table(self):
        return self.get_table('sj_sesson')

    def get_sj_url_table(self):
        return self.get_table('sj_url')

    def get_sj_url_tmp_table(self):
        return self.get_table('sj_url_tmp')

    def get_sj_user_table(self):
        return self.get_table('sj_user')

    def get_sj_video_table(self):
        return self.get_table('sj_video')

if __name__ == '__main__':
    db = MigrateDB()
    db.connect()
    for row in db.get_sj_user_table():
        for key, value in row.iteritems():
             print key, ', ', value, ', ', type(value)
        print