def test_database_layer_nulldriver(self):
# Define models by calling the class
node = Node()
node.name = "My awsome node"
# Add 1 port
node.append_port(name="port1",
direction="in",
protocol="ros1",
parameters=[],
callback={
"_id":
"",
"_name":
"my awsome callback",
"tags": [{
"name": "label",
"value": "my awsome callback"
}],
"parameters": [{
"name": "ros_parameters",
"data": {
"msgtype": "movaimsg",
"scene": "scene_a"
}
}],
"code":
"print(globals())",
"libraries": []
})
# Checking num of ports
self.assertEqual(len(node.ports), 1)
# Checking port content
self.assertEqual(node.ports[0].name, "port1")
self.assertEqual(node.ports[0].direction, "in")
self.assertEqual(node.ports[0].protocol, "ros1")
self.assertEqual(len(node.ports[0].parameters), 0)
db = DatabaseLayer(drv=RedisDriver())
ids = db.store(node)
node.version = "1.1"
ids = db.store(node)
class UrlHandler(object):
"""Main handler for the parser. Checks the given URL and
decides if it should be written too the JSON file or passed."""
def __init__(self, url):
self.url = urlparse(url)
self.raw_url = url
self.db = DatabaseLayer()
self.json = json.loads(open("settings.json", "rb").read())
def valid_link(self, link):
"""Checks if the given URL is already read.
Input:
link:str = URL
Return:
bool True or False"""
if link.encode('utf-8') in self.db.fetch():
return False
a = urlparse(link)
if a.netloc in self.json[self.url.netloc]["links"]:
if "http" in link:
if a.path == "" or a.path == "/":
self.link = a.geturl()
else:
self.link = a.path.rsplit(".", 1)[0]
if self.link[-1] == "/": self.link = self.link[:-1]
if "article" in self.link:
if not a.path.rsplit("/", 1)[0] == "":
self.link = a.path.rsplit("/", 1)[0]
else:
l = link.find("article")
self.link = link[:l+len("article")]
if not self._tags(self.link):
return False
return True
def _tags(self, tag):
list_tags = self.json[self.url.netloc]["tags"]
for i in list_tags:
if i == tag:
return False
if "*" in i.split("/"):
if i.rsplit("/", 1)[0] in tag:
return False
return True
def url_write(self, link):
"""Checks the URL if valid or not.
Then writes too file and 'database'
depending on the result.
Input:
link:str = URL
Return:
bool True or False depending on result"""
if self.valid_link(link):
try:
with open(os.getcwd()+"/tmp/output.json", "rb") as f:
self.output = json.loads(f.read(),encoding="utf-8")
except:
self.output = {}
try:
self.output[self.url.netloc][self.link] += 1
except Exception, e:
if self.output == {} or self.url.netloc not in self.output.keys():
self.output[self.url.netloc] = {}
self.output[self.url.netloc][self.link] = 0
self.output[self.url.netloc][self.link] += 1
self.db.insert(link)
self._output_stuff()
return True
else:
def __init__(self, url):
self.url = urlparse(url)
self.raw_url = url
self.db = DatabaseLayer()
self.json = json.loads(open("settings.json", "rb").read())
