|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 # Copyright (c) 2012 - 2021 Detlev Offenbach <detlev@die-offenbachs.de> |
|
4 # |
|
5 |
|
6 """ |
|
7 Module implementing a class to write login data files. |
|
8 """ |
|
9 |
|
10 from PyQt5.QtCore import QXmlStreamWriter, QIODevice, QFile |
|
11 |
|
12 |
|
13 class PasswordWriter(QXmlStreamWriter): |
|
14 """ |
|
15 Class implementing a writer object to generate login data files. |
|
16 """ |
|
17 def __init__(self): |
|
18 """ |
|
19 Constructor |
|
20 """ |
|
21 super().__init__() |
|
22 |
|
23 self.setAutoFormatting(True) |
|
24 |
|
25 def write(self, fileNameOrDevice, logins, forms, nevers): |
|
26 """ |
|
27 Public method to write an login data file. |
|
28 |
|
29 @param fileNameOrDevice name of the file to write (string) |
|
30 or device to write to (QIODevice) |
|
31 @param logins dictionary with login data (user name, password) |
|
32 @param forms list of forms data (list of LoginForm) |
|
33 @param nevers list of URLs to never store data for (list of strings) |
|
34 @return flag indicating success (boolean) |
|
35 """ |
|
36 if isinstance(fileNameOrDevice, QIODevice): |
|
37 f = fileNameOrDevice |
|
38 else: |
|
39 f = QFile(fileNameOrDevice) |
|
40 if not f.open(QFile.WriteOnly): |
|
41 return False |
|
42 |
|
43 self.setDevice(f) |
|
44 return self.__write(logins, forms, nevers) |
|
45 |
|
46 def __write(self, logins, forms, nevers): |
|
47 """ |
|
48 Private method to write an login data file. |
|
49 |
|
50 @param logins dictionary with login data (user name, password) |
|
51 @param forms list of forms data (list of LoginForm) |
|
52 @param nevers list of URLs to never store data for (list of strings) |
|
53 @return flag indicating success (boolean) |
|
54 """ |
|
55 self.writeStartDocument() |
|
56 self.writeDTD("<!DOCTYPE passwords>") |
|
57 self.writeStartElement("Password") |
|
58 self.writeAttribute("version", "2.0") |
|
59 |
|
60 if logins: |
|
61 self.__writeLogins(logins) |
|
62 if forms: |
|
63 self.__writeForms(forms) |
|
64 if nevers: |
|
65 self.__writeNevers(nevers) |
|
66 |
|
67 self.writeEndDocument() |
|
68 return True |
|
69 |
|
70 def __writeLogins(self, logins): |
|
71 """ |
|
72 Private method to write the login data. |
|
73 |
|
74 @param logins dictionary with login data (user name, password) |
|
75 """ |
|
76 self.writeStartElement("Logins") |
|
77 for key, login in logins.items(): |
|
78 self.writeEmptyElement("Login") |
|
79 self.writeAttribute("key", key) |
|
80 self.writeAttribute("user", login[0]) |
|
81 self.writeAttribute("password", login[1]) |
|
82 self.writeEndElement() |
|
83 |
|
84 def __writeForms(self, forms): |
|
85 """ |
|
86 Private method to write forms data. |
|
87 |
|
88 @param forms list of forms data (list of LoginForm) |
|
89 """ |
|
90 self.writeStartElement("Forms") |
|
91 for key, form in forms.items(): |
|
92 self.writeStartElement("Form") |
|
93 self.writeAttribute("key", key) |
|
94 self.writeAttribute("url", form.url.toString()) |
|
95 self.writeAttribute("name", str(form.name)) |
|
96 self.writeTextElement("PostData", form.postData) |
|
97 self.writeEndElement() |
|
98 self.writeEndElement() |
|
99 |
|
100 def __writeNevers(self, nevers): |
|
101 """ |
|
102 Private method to write the URLs never to store login data for. |
|
103 |
|
104 @param nevers list of URLs to never store data for (list of strings) |
|
105 """ |
|
106 self.writeStartElement("Nevers") |
|
107 for never in nevers: |
|
108 self.writeEmptyElement("Never") |
|
109 self.writeAttribute("url", never) |
|
110 self.writeEndElement() |