add: initial AI support
This commit is contained in:
diSabler
180
scripts/update2_xto3_0.py
Normal file
180
scripts/update2_xto3_0.py
Normal file
@@ -0,0 +1,180 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# --------------------------------------------------------------------------- #
|
||||
# #
|
||||
# Update iSida Jabber Bot from version 2.x to 3.x #
|
||||
# Copyright (C) 2012 diSabler <dsy@dsy.name> #
|
||||
# #
|
||||
# This program is free software: you can redistribute it and/or modify #
|
||||
# it under the terms of the GNU General Public License as published by #
|
||||
# the Free Software Foundation, either version 3 of the License, or #
|
||||
# (at your option) any later version. #
|
||||
# #
|
||||
# This program is distributed in the hope that it will be useful, #
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
||||
# GNU General Public License for more details. #
|
||||
# #
|
||||
# You should have received a copy of the GNU General Public License #
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. #
|
||||
# #
|
||||
# --------------------------------------------------------------------------- #
|
||||
|
||||
import os, sys, sqlite3, psycopg2, re, time
|
||||
|
||||
os.chdir('..')
|
||||
|
||||
set_folder = u'settings/%s'
|
||||
|
||||
agestat = set_folder % 'agestat.db'
|
||||
jidbase = set_folder % 'jidbase.db'
|
||||
talkers = set_folder % 'talkers.db'
|
||||
wtfbase = set_folder % 'wtfbase2.db'
|
||||
answers = set_folder %'answers.db'
|
||||
saytobase = set_folder % 'sayto.db'
|
||||
distbase = set_folder % 'dist.db'
|
||||
distbase_user = set_folder % 'dist_user.db'
|
||||
karmabase = set_folder % 'karma.db'
|
||||
acl_base = set_folder %'acl.db'
|
||||
wzbase = set_folder % 'wz.db'
|
||||
gisbase = set_folder % 'gis.db'
|
||||
def_phone_base = set_folder % 'defcodes.db'
|
||||
muc_lock_base = set_folder % 'muclock.db'
|
||||
|
||||
configname = 'settings/config.py'
|
||||
|
||||
def convert_base(filename,from_base,to_base):
|
||||
global conn
|
||||
if os.path.isfile(filename):
|
||||
s_conn = sqlite3.connect(filename)
|
||||
s_cur = s_conn.cursor()
|
||||
tmp = s_cur.execute('select * from %s' % from_base).fetchall()
|
||||
print 'Import %s, Base %s->%s, %s records' % (filename, from_base, to_base, len(tmp))
|
||||
s_conn.close()
|
||||
if tmp:
|
||||
for t in tmp:
|
||||
req = 'insert into %s values(%s);' % (to_base,','.join(['%'+a for a in 's'*len(t)]))
|
||||
cur.execute(req,t)
|
||||
conn.commit()
|
||||
|
||||
def errorHandler(text):
|
||||
print text
|
||||
sys.exit()
|
||||
|
||||
print 'Updater for Isida Jabber Bot from 2.x to 3.x'
|
||||
print '(c) Disabler Production Lab.'
|
||||
|
||||
base_charset, base_type = 'utf8', 'pgsql'
|
||||
|
||||
if os.path.isfile(configname): execfile(configname)
|
||||
else: errorHandler('%s is missed.' % configname)
|
||||
|
||||
try: _,_,_,_,_ = base_name,base_user,base_host,base_pass,base_port
|
||||
except: errorHandler('Missed settings for SQL DB!')
|
||||
|
||||
if base_type == 'pgsql':
|
||||
import psycopg2
|
||||
import psycopg2.extensions
|
||||
conn = psycopg2.connect(database=base_name, user=base_user, host=base_host, password=base_pass, port=base_port)
|
||||
elif base_type == 'mysql':
|
||||
import MySQLdb
|
||||
conn = MySQLdb.connect(db=base_name, user=base_user, host=base_host, passwd=base_pass, port=int(base_port), charset=base_charset)
|
||||
else: errorHandler('Unknown database backend!')
|
||||
|
||||
print 'Base type: %s' % base_type
|
||||
|
||||
cur = conn.cursor()
|
||||
|
||||
al = [t.lower() for t in sys.argv]
|
||||
|
||||
if '/wtf_update' in al:
|
||||
print 'Update `wtf` base'
|
||||
cur.execute('select * from wtf;')
|
||||
w = cur.fetchall()
|
||||
cur.execute('drop table wtf;')
|
||||
cur.execute('CREATE TABLE wtf (\
|
||||
ind integer,\
|
||||
room text,\
|
||||
jid text,\
|
||||
nick text,\
|
||||
wtfword text,\
|
||||
wtftext text,\
|
||||
time integer,\
|
||||
lim integer);')
|
||||
for tmp in w:
|
||||
try: tm = time.mktime([[int(t) for t in tmp[6].replace('.',' ').replace(':',' ').split()][t1] for t1 in [2,1,0,3,4,5]]+[0,0,0])
|
||||
except: tm = 0
|
||||
r = tmp[:6]+(tm,)+tmp[7:]
|
||||
req = 'insert into wtf values(%s);' % (','.join(['%'+a for a in 's'*len(r)]))
|
||||
cur.execute(req,r)
|
||||
print 'Updated %s record(s)' % len(w)
|
||||
print 'Finished!'
|
||||
|
||||
elif '/gis_update' in al:
|
||||
print 'Update `gis` base'
|
||||
cur.execute('delete from gis;')
|
||||
from_base = to_base = 'gis'
|
||||
s_conn = sqlite3.connect(gisbase)
|
||||
s_cur = s_conn.cursor()
|
||||
tmp = s_cur.execute('select * from %s' % from_base).fetchall()
|
||||
print 'Import %s, Base %s->%s, %s records' % (gisbase, from_base, to_base, len(tmp))
|
||||
s_conn.close()
|
||||
|
||||
if tmp:
|
||||
for t in tmp:
|
||||
req = 'insert into gis values(%s);' % (','.join(['%'+a for a in 's'*len(t)]))
|
||||
cur.execute(req,t)
|
||||
print 'Updated %s record(s)' % len(tmp)
|
||||
else: 'Update failed! `gis` base not found or empty!'
|
||||
print 'Finished!'
|
||||
|
||||
elif '/h' in al or '/help' in al or '--help' in al:
|
||||
print '\
|
||||
Usage: python %s [/h|/wtf_update|/gis_update]\n\n\
|
||||
/h - this help\n\
|
||||
/wtf_update - update number format in `wtf` base for enable history of words definitions\n\
|
||||
' % al[0]
|
||||
elif len(al) > 1: print 'Use: python %s /h for short help' % al[0]
|
||||
else:
|
||||
bases_arr = [[jidbase,'jid','jid'],[talkers,'talkers','talkers'],
|
||||
[answers,'answer','answer'],[saytobase,'st','sayto'],[distbase,'dist','dist'],[distbase_user,'dist','dist_user'],
|
||||
[karmabase,'karma','karma'],[karmabase,'commiters','karma_commiters'],[karmabase,'limits','karma_limits'],
|
||||
[acl_base,'acl','acl'],[wzbase,'wz','wz'],[gisbase,'gis','gis'],[def_phone_base,'ru_stat','def_ru_stat'],
|
||||
[def_phone_base,'ua_stat','def_ua_stat'],[def_phone_base,'ru_mobile','def_ru_mobile'],[muc_lock_base,'muc','muc_lock']]
|
||||
|
||||
from_base = to_base = 'age'
|
||||
if os.path.isfile(agestat):
|
||||
s_conn = sqlite3.connect(agestat)
|
||||
s_cur = s_conn.cursor()
|
||||
tmp = s_cur.execute('select * from %s' % from_base).fetchall()
|
||||
print 'Import %s, Base %s->%s, %s records' % (agestat, from_base, to_base, len(tmp))
|
||||
s_conn.close()
|
||||
if tmp:
|
||||
for t in tmp:
|
||||
req = 'insert into %s values(%s);' % (to_base,','.join(['%'+a for a in 's'*9]))
|
||||
cur.execute(req,list(t)+[t[1].lower()])
|
||||
conn.commit()
|
||||
|
||||
from_base = to_base = 'wtf'
|
||||
if os.path.isfile(wtfbase):
|
||||
s_conn = sqlite3.connect(wtfbase)
|
||||
s_cur = s_conn.cursor()
|
||||
tmp = s_cur.execute('select * from %s' % from_base).fetchall()
|
||||
print 'Import %s, Base %s->%s, %s records' % (wtfbase, from_base, to_base, len(tmp))
|
||||
s_conn.close()
|
||||
if tmp:
|
||||
for t in tmp:
|
||||
try: tm = time.mktime([[int(t2) for t2 in t[6].replace('.',' ').replace(':',' ').split()][t1] for t1 in [2,1,0,3,4,5]]+[0,0,0])
|
||||
except: tm = 0
|
||||
r = t[:6]+(tm,)+t[7:]
|
||||
req = 'insert into %s values(%s);' % (to_base,','.join(['%'+a for a in 's'*len(r)]))
|
||||
cur.execute(req,r)
|
||||
conn.commit()
|
||||
|
||||
for t in bases_arr: convert_base(t[0],t[1],t[2])
|
||||
print 'Finished!'
|
||||
|
||||
cur.close()
|
||||
conn.commit()
|
||||
conn.close()
|
||||
Reference in New Issue
Block a user