1 files changed, 80 insertions, 0 deletions
diff --git a/hercules/code/server/party.py b/hercules/code/server/party.py
new file mode 100644
index 0000000..c75a541
--- /dev/null
+++ b/hercules/code/server/party.py
@@ -0,0 +1,80 @@
+# -*- 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")