Stationeers_Localization_Tool/transfer_localization.py at master · FliesWithWind/Stationeers_Localization_Tool · GitHub

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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import xml.etree.ElementTree as ET
import argparse


#parse command line input
PARSER = argparse.ArgumentParser(description=
								"""
								This script imports tranlsated strings from your old XML localization file.
								ex. "python transfer_localization.py -o polish.xml -n english.xml"
								""")
PARSER.add_argument('-o', '--oldfile', help=
					"""
					Old localization xml file.
					"""
					, required=True)
PARSER.add_argument('-n', '--newfile', help=
					"""
					New english localizatiion xml file.
					"""
					, default='english.xml', required=False)
PARSER.add_argument('-c','--combine', action='store_true', default=False, help=
					"""
					Cobines old translation with newly translated strings.
					""", required=False)
PARSER.add_argument('-t', '--translation', help=
					"""
					Name of extracted file that you translated. By default "newstrings.xml".
					"""
					, default='newstrings.xml', required=False)

ARGS = PARSER.parse_args()


def extract():
	new = ET.parse(ARGS.newfile)
	old = ET.parse(ARGS.oldfile)
	newroot = new.getroot()
	oldroot = old.getroot()

	for child in newroot:
		for record_reagent in oldroot.iter('RecordReagent'):
			oldkey = record_reagent.find('Key').text
			for new_record_reagent in child.iter('RecordReagent'):
				newkey = new_record_reagent.find('Key').text
				if newkey == oldkey:
					child.remove(new_record_reagent)

		for record in oldroot.iter('Record'):
			oldkey = record.find('Key').text
			for new_record in child.iter('Record'):
				found = False
				newkey = new_record.find('Key').text
				if newkey == oldkey:
					child.remove(new_record)

	new.write('newstrings.xml')

def importer(oldroot, newroot):
	for record_reagent in oldroot.iter('RecordReagent'):
		oldkey = record_reagent.find('Key').text
		oldvalue = record_reagent.find('Value').text
		for new_record_reagent in newroot.iter('RecordReagent'):
			newkey = new_record_reagent.find('Key').text
			newvalue = new_record_reagent.find('Value').text
			value = new_record_reagent.find('Value')

			if newkey == oldkey:
				value.text = oldvalue
				unit = new_record_reagent.find('Unit')
				oldunit = record_reagent.find('Unit').text
				unit.text = oldunit


	for record in oldroot.iter('Record'):
		oldkey = record.find('Key').text
		oldvalue = record.find('Value').text
		for new_record in newroot.iter('Record'):
			newkey = new_record.find('Key').text
			newvalue = new_record.find('Value').text
			value = new_record.find('Value')
			if newkey == oldkey:
				value.text = oldvalue

def combine():
	#new translation file english.xml
	new = ET.parse(ARGS.newfile)
	newroot = new.getroot()
	#old translation file
	old = ET.parse(ARGS.oldfile)
	oldroot = old.getroot()
	#newstrings.xml
	trs = ET.parse(ARGS.translation)
	trsroot = trs.getroot()

	#Copying language Name, Code and Font
	oldname = oldroot.find('Name')
	newname = newroot.find('Name')
	newname.text = oldname.text
	oldcode = oldroot.find('Code')
	newcode = newroot.find('Code')
	newcode.text = oldcode.text
	oldfont = oldroot.find('Font')
	newfont = newroot.find('Font')
	newfont.text = oldfont.text

	#Importing
	importer(oldroot,newroot)
	importer(trsroot,newroot)

	newroot.set('xmlns:xsi', 'http://www.w3.org/2001/XMLSchema-instance')
	newroot.set('xmlns:xsd', 'http://www.w3.org/2001/XMLSchema')

	new.write('output.xml',encoding="utf-8", xml_declaration=True)

if ARGS.combine:
	combine()
else:
	extract()