# -*- coding: utf8 -*-
#
# Copyright (C) 2014 Evol Online
# Author: Andrei Karas (4144)
import re
from code.fileutils import readFile
from code.stringutils import stripNewLine, escapeSqlStr
def findLeaderId(name, users):
for userId in users:
user = users[userId]
if user.char_name == name:
return user.char_id
return 0
def convertParty(users):
srcFile = "olddb/party.txt"
dstFile = "newdb/party.sql"
fieldsSplit = re.compile("\t")
comaSplit = re.compile(",")
tpl = readFile("templates/party.sql")
firstLine = True
with open(dstFile, "w") as w:
w.write(tpl)
w.write("INSERT INTO `party` VALUES ")
with open(srcFile, "r") as r:
for line in r:
if line[:2] == "//":
continue
line = stripNewLine(line)
rows = fieldsSplit.split(line)
if len(rows) == 2:
continue
if len(rows) < 3:
print "wrong party.txt line: " + line
continue
partyId = rows[0]
partyName = rows[1]
tmp = comaSplit.split(rows[2])
partyExp = tmp[0]
partyItem = tmp[1]
leaderId = 0
leaderName = ""
accountId = ""
for f in xrange(3, len(rows), 2):
if rows[f] == "0,0" or rows[f] == "":
continue
tmp = comaSplit.split(rows[f])
accountId = tmp[0]
leader = tmp[1]
charName = rows[f + 1]
if leader == "1":
leaderId = accountId
leaderName = charName
if firstLine == False:
w.write(",\n")
else:
firstLine = False
leaderCharId = findLeaderId(leaderName, users)
w.write(("({party_id},'{name}',{exp},{item}," +
"{leader_id},{leader_char})").format(
party_id = partyId,
name = escapeSqlStr(partyName),
exp = partyExp,
item = partyItem,
leader_id = leaderId,
leader_char = leaderCharId
))
w.write("\n")
