initial commit

2019-12-23 12:42:31 +08:00
parent 30458acfd8
commit 1ac6d0bb9c
22 changed files with 2126 additions and 1 deletions
--- a/subscriber_utils.py
+++ b/subscriber_utils.py
@@ -0,0 +1,92 @@
+from lib.parser import dom_node, simple_parser
+
+
+class subscribe_manager():
+    def __init__(self, subscriber_config = './config/subscriber.xml'):
+        self.subscriber_config = None
+        self.subscribers = {}
+        if subscriber_config is not None:
+            self.subscriber_config  = subscriber_config
+            self.load()
+
+    def show(self):
+        if self.subscribers is None:
+            print('No subscriber found!')
+        else:
+            for name in self.subscribers:
+                print('Name:', name, 'Email:', self.subscribers[name]['email'])
+
+    def load(self, path=None):
+        if path is None:
+            path = self.subscriber_config
+        if path is None:
+            return None
+        tree = None
+        with open(path, 'r') as f:
+            xml = f.read()
+            parser = simple_parser()
+            parser.feed(xml)
+            tree = parser.root
+        subscribers = {}
+        if tree is not None:
+            for person in tree.childs:
+                person_name = None
+                person_email = None
+                person_topics = []
+                person_keywords = []
+                for item in person.childs:
+                    if item.name == 'name':
+                        person_name = item.data
+                    elif item.name == 'email':
+                        person_email = item.data
+                    elif item.name == 'topics':
+                        for topic in item.childs:
+                            if topic.name == 'topic':
+                                person_topics.append(topic.data)
+                    elif item.name == 'keywords':
+                        for keyword in item.childs:
+                            if keyword.name == 'keyword':
+                                person_keywords.append(keyword.data)
+                if person_name is not None and person_email is not None and person_topics is not None:
+                    subscriber = {}
+                    subscriber['keywords'] = person_keywords
+                    subscriber['email'] = person_email
+                    subscriber['topics'] = person_topics
+                    subscribers[person_name] = subscriber
+        self.subscribers = subscribers
+        print('Subscriber load success! All subscribers are shown below:')
+        self.show();
+
+    def get_subscribed_topics(self):
+        topics = []
+        for name in self.subscribers:
+            subscriber = self.subscribers[name]
+            topics += subscriber['topics']
+        topics = set(topics)
+        return topics
+
+    def get_subscribed_keywords(self):
+        keywords = []
+        for name in self.subscribers:
+            keywords += self.subscribers[name]['keywords']
+        keywords = set(keywords)
+        return keywords
+
+    def get_keywords_of_topics(self):
+        keywords_of_topics = {}
+        for name in self.subscribers:
+            subscriber = self.subscribers[name]
+            topic_group = subscriber['topics']
+            for topic in topic_group:
+                if topic not in keywords_of_topics:
+                    keywords_of_topics[topic] = []
+                keywords_of_topics[topic] += subscriber['keywords']
+        return keywords_of_topics
+
+
+if __name__ == '__main__':
+    manager = subscribe_manager()
+    print(manager.subscribers)
+    print(manager.get_subscribed_topics())
+    print(manager.get_subscribed_keywords())
+    print(manager.get_keywords_of_topics())