-
Notifications
You must be signed in to change notification settings - Fork 0
/
node.py
27 lines (24 loc) · 863 Bytes
/
node.py
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
class Node:
def __init__(self, container, element, class_name):
self.container = container
self.element = element
self.className = class_name
def get_text(self):
curr_node = self.container.find(self.element, class_=self.className)
if curr_node is not None:
return curr_node.text
else:
return None
def get_link(self):
curr_node = self.container.find(self.element, class_=self.className)
if curr_node is not None:
return curr_node["href"]
else:
return None
def get_email(self):
into_container = self.container.find(self.element, class_=self.className)
infos = into_container.findAll("li")
for info in infos:
if "@" in info.text:
return info.text.strip()
return None