diff options
Diffstat (limited to 'tools/migrate_db.py')
| -rwxr-xr-x | tools/migrate_db.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/tools/migrate_db.py b/tools/migrate_db.py new file mode 100755 index 0000000..16bf30c --- /dev/null +++ b/tools/migrate_db.py @@ -0,0 +1,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 |
