dataimport.py
changeset 6492 47a284c0d012
parent 6427 c8a5ac2d1eaa
child 6989 4a999a647f52
--- a/dataimport.py	Thu Oct 14 10:15:38 2010 +0200
+++ b/dataimport.py	Thu Oct 14 10:15:56 2010 +0200
@@ -85,7 +85,7 @@
 
 def count_lines(stream_or_filename):
     if isinstance(stream_or_filename, basestring):
-        f = open(filename)
+        f = open(stream_or_filename)
     else:
         f = stream_or_filename
         f.seek(0)
@@ -98,8 +98,8 @@
                   skipfirst=False, withpb=True):
     """same as ucsvreader but a progress bar is displayed as we iter on rows"""
     if isinstance(stream_or_path, basestring):
-        if not osp.exists(filepath):
-            raise Exception("file doesn't exists: %s" % filepath)
+        if not osp.exists(stream_or_path):
+            raise Exception("file doesn't exists: %s" % stream_or_path)
         stream = open(stream_or_path)
     else:
         stream = stream_or_path
@@ -421,7 +421,6 @@
         ObjectStore.__init__(self)
         if session is None:
             sys.exit('please provide a session of run this script with cubicweb-ctl shell and pass cnx as session')
-            session = cnx
         if not hasattr(session, 'set_pool'):
             # connection
             cnx = session