From 93a8a9c6647ec22deffb43cf2b99183b6a0e9c9a Mon Sep 17 00:00:00 2001
From: davebshow <davebshow@gmail.com>
Date: Sun, 3 Jul 2016 02:20:06 -0400
Subject: [PATCH] polishing
---
goblin/api.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/goblin/api.py b/goblin/api.py
index 1528da1..8086bf5 100644
--- a/goblin/api.py
+++ b/goblin/api.py
@@ -99,12 +99,15 @@ class Session:
while self._pending:
elem = self._pending.popleft()
await self.save_element(elem)
-
+
async def save_element(self, element):
if element.__type__ == 'vertex':
- await self.save_vertex(element)
+ result = await self.save_vertex(element)
elif element.__type__ == 'edge':
- await self.save_edge(element)
+ result = await self.save_edge(element)
+ else:
+ raise Exception("Unkown element type")
+ return result
async def save_vertex(self, element):
result = await self._save_element(element,
--
GitLab