summaryrefslogtreecommitdiff
path: root/tools/migrate_db.py
blob: 16bf30cdd2a923a9a36891b448ad052eeb0dac53 (plain)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/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