Refatoração do codigo

marcydoty · Dec 28, 2010 · 01c2bc7 · 01c2bc7
1 parent f44cb98
commit 01c2bc7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 20 deletions.
diff --git a/src/.swp b/src/.swp
diff --git a/src/correios.py b/src/correios.py
@@ -18,11 +18,21 @@
 
 import cod
 
+def __getDados(tags_name, dom):
+    dados = {}
+
+    for tag_name in tags_name:
+        try:
+            dados[tag_name] = dom.getElementsByTagName(tag_name)[0]
+            dados[tag_name] = dados[tag_name].childNodes[0].data
+        except:
+            dados[tag_name] = ''
+
+    return dados
+
 def frete(cod,GOCEP,HERECEP,peso,
           comprimento,diametro,toback='xml'):
 
-    dados = {}
-
     url = "http://ws.correios.com.br/calculador/CalcPrecoPrazo.aspx?\
            StrRetorno=%s\
            &nCdServico=%s\
@@ -45,18 +55,9 @@ def frete(cod,GOCEP,HERECEP,peso,
                  'EntregaSabado',
                 )
 
-    for tag_name in tags_name:
-        try:
-            dados[tag_name] = dom.getElementsByTagName(tag_name)[0]
-            dados[tag_name] = dados[tag_name].childNodes[0].data
-        except:
-            dados[tag_name] = ''
-
-    return dados
+    return __getDados(tags_name, dom)
 
 def cep(numero):
-    dados = {}
-
     url = 'http://cep.republicavirtual.com.br/web_cep.php?\
            formato=xml&cep=%s' % (str(numero))
     dom = minidom.parse(urllib2.urlopen(url))
@@ -71,14 +72,9 @@ def cep(numero):
     resultado = dom.getElementsByTagName('resultado')[0]
     resultado = int(resultado.childNodes[0].data)
     if resultado != 0:
-        for tag_name in tags_name:
-            try:
-                dados[tag_name] = dom.getElementsByTagName(tag_name)[0]
-                dados[tag_name] = dados[tag_name].childNodes[0].data
-            except:
-                dados[tag_name] = ''
-
-    return dados
+        return __getDados(tags_name, dom)
+    else:
+        return {}
 
 # Delete the modules to not show up in the namespace
 # del  urllib2, sys, minidom