SimpleHTMLUpdateChecker/UpdateFileReader.py at master · e-marco/SimpleHTMLUpdateChecker · 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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
import UpdateItem as ui
import os.path
from shutil import copyfile

class UpdateFileReader:
	def __init__(self, filename, is_verbose):
		self.status = True
		self.is_verbose = is_verbose
		self.filename = filename
		self.currentlineindex = 0

		if len(filename) > 0:
			self.filename = filename
		else:
			self.status = False
			warning_info = "Empty filename"

		if self.status:
			if os.path.isfile(self.filename) == False:
				self.status = False
				warning_info = "File does not exist"

		if self.status:
			self.message = "Info: UpdateFileReader initialization okay"
		else:
			self.message = "Warning: " + warning_info

		if self.is_verbose:
			print(self.message)

	def getNextItem(self):
		if self.status != True:
			if self.is_verbose:
				print("Warning: UpdateFileReader not initialized")
			return None
		else:
			try:
				fileHandle = open(self.filename, 'r', encoding='utf-8')
				lines = fileHandle.readlines()
				fileHandle.close()
			except:
				self.status = False
				self.message = "File could not be read."
				if self.is_verbose:
					print(self.message)
				return None

			if self.currentlineindex < len(lines):
				current_line = lines[self.currentlineindex]
				current_line = current_line.strip()
				self.currentlineindex += 1
				return True
			else:
				return None

	def getCurrentItemData(self):
		if self.status != True:
			if self.is_verbose:
				print("Warning: UpdateFileReader not initialized")
			return None
		else:
			try:
				fileHandle = open(self.filename, 'r', encoding='utf-8')
				lines = fileHandle.readlines()
				fileHandle.close()
			except:
				self.status = False
				self.message = "File could not be read."
				if self.is_verbose:
					print(self.message)
				return None

			if self.currentlineindex <= len(lines):
				current_line = lines[self.currentlineindex-1]
				current_item = extractData(current_line, self.is_verbose)

				if current_item is None:
					self.status = False
					self.message = "Error: Reading info for " + current_line
				else:
					self.message = "Info: Reading info for " + current_item.name + " okay"

				if self.is_verbose:
					print(self.message)

				if self.status:
					return current_item
				else:
					return None
			else:
				return None

	def updateCurrentItemData(self, UpdateItem):
		if self.status != True:
			if self.is_verbose:
				print("Warning: UpdateFileReader not initialized")
			return None
		else:
			try:
				fileHandle = open(self.filename, 'r', encoding='utf-8')
				lines = fileHandle.readlines()
				fileHandle.close()
			except:
				self.status = False
				self.message = "File could not be read."
				if self.is_verbose:
					print(self.message)
				return None

			if self.currentlineindex <= len(lines):
				if UpdateItem.status:
					# create backup
					copyfile(self.filename,"backup.tmp")
					try:
						lines[self.currentlineindex-1] = UpdateItem.name + ";" + UpdateItem.url + ";" + UpdateItem.re_pattern + ";" + UpdateItem.installed_version + "\n"
						fileHandle = open(self.filename, 'w', encoding='utf-8')
						fileHandle.writelines(lines)
						fileHandle.close()
						os.remove("backup.tmp")
						self.message = "Info: File updated successfully"
					except:
						self.status = False
						self.message = "Error: Writing file. Please use backup.tmp"

				else:
					self.status = False
					self.message = "Warning: UpdateItem invalid"

				if self.is_verbose:
					print(self.message)
				if self.status:
					return self
				else:
					return None
			else:
				return None

def extractData(line, is_verbose):
	try:
		line = line.strip()
		current_data = line.split(';')
		current_item = ui.UpdateItem(current_data[0], current_data[1], current_data[2], current_data[3], is_verbose)
		return current_item
	except:
		message = "Check file format."
		if is_verbose:
			print(message)
		return None