Mercurial Repositories > eric / file comparison
comparison: eric7/WebBrowser/UserAgent/UserAgentReader.py
eric7/WebBrowser/UserAgent/UserAgentReader.py
- branch
- eric7
- changeset 8312
- 800c432b34c8
- parent 8218
- 7c09585bd960
- child 8318
- 962bce857696
equal
deleted
inserted
replaced
| 8311:4e8b98454baa | 8312:800c432b34c8 |
|---|---|
| 1 # -*- coding: utf-8 -*- | |
| 2 | |
| 3 # Copyright (c) 2012 - 2021 Detlev Offenbach <detlev@die-offenbachs.de> | |
| 4 # | |
| 5 | |
| 6 | |
| 7 """ | |
| 8 Module implementing a class to read user agent data files. | |
| 9 """ | |
| 10 | |
| 11 from PyQt5.QtCore import QXmlStreamReader, QIODevice, QFile, QCoreApplication | |
| 12 | |
| 13 | |
| 14 class UserAgentReader(QXmlStreamReader): | |
| 15 """ | |
| 16 Class implementing a reader object for user agent data files. | |
| 17 """ | |
| 18 def __init__(self): | |
| 19 """ | |
| 20 Constructor | |
| 21 """ | |
| 22 super().__init__() | |
| 23 | |
| 24 def read(self, fileNameOrDevice): | |
| 25 """ | |
| 26 Public method to read a user agent file. | |
| 27 | |
| 28 @param fileNameOrDevice name of the file to read (string) | |
| 29 or reference to the device to read (QIODevice) | |
| 30 @return dictionary with user agent data (host as key, agent string as | |
| 31 value) | |
| 32 """ | |
| 33 self.__agents = {} | |
| 34 | |
| 35 if isinstance(fileNameOrDevice, QIODevice): | |
| 36 self.setDevice(fileNameOrDevice) | |
| 37 else: | |
| 38 f = QFile(fileNameOrDevice) | |
| 39 if not f.exists(): | |
| 40 return self.__agents | |
| 41 f.open(QFile.ReadOnly) | |
| 42 self.setDevice(f) | |
| 43 | |
| 44 while not self.atEnd(): | |
| 45 self.readNext() | |
| 46 if self.isStartElement(): | |
| 47 version = self.attributes().value("version") | |
| 48 if ( | |
| 49 self.name() == "UserAgents" and | |
| 50 (not version or version == "1.0") | |
| 51 ): | |
| 52 self.__readUserAgents() | |
| 53 else: | |
| 54 self.raiseError(QCoreApplication.translate( | |
| 55 "UserAgentReader", | |
| 56 "The file is not a UserAgents version 1.0 file.")) | |
| 57 | |
| 58 return self.__agents | |
| 59 | |
| 60 def __readUserAgents(self): | |
| 61 """ | |
| 62 Private method to read the user agents data. | |
| 63 """ | |
| 64 if not self.isStartElement() and self.name() != "UserAgents": | |
| 65 return | |
| 66 | |
| 67 while not self.atEnd(): | |
| 68 self.readNext() | |
| 69 if self.isEndElement(): | |
| 70 if self.name() == "UserAgent": | |
| 71 continue | |
| 72 else: | |
| 73 break | |
| 74 | |
| 75 if self.isStartElement(): | |
| 76 if self.name() == "UserAgent": | |
| 77 attributes = self.attributes() | |
| 78 host = attributes.value("host") | |
| 79 agent = attributes.value("agent") | |
| 80 self.__agents[host] = agent | |
| 81 else: | |
| 82 self.__skipUnknownElement() | |
| 83 | |
| 84 def __skipUnknownElement(self): | |
| 85 """ | |
| 86 Private method to skip over all unknown elements. | |
| 87 """ | |
| 88 if not self.isStartElement(): | |
| 89 return | |
| 90 | |
| 91 while not self.atEnd(): | |
| 92 self.readNext() | |
| 93 if self.isEndElement(): | |
| 94 break | |
| 95 | |
| 96 if self.isStartElement(): | |
| 97 self.__skipUnknownElement() |
