Changeset 154

Show
Ignore:
Timestamp:
12/08/09 18:10:25 (5 years ago)
Author:
vinicius
Message:

Small refactoring on lastID call

Location:
donkeysurvey/database
Files:
10 modified

Legend:

Unmodified
Added
Removed
  • donkeysurvey/database/AddressDAO.py

    r145 r154  
    2424                logging.debug(queryInsert) 
    2525                self.cursor.execute(queryInsert) 
    26                 last = CommonDAO.lastID(self, self.tablename) 
     26                last = self.lastID(self.tablename) 
    2727                return last 
    2828        except Exception, err: 
  • donkeysurvey/database/AddressHasFileDAO.py

    r151 r154  
    2121                logging.debug(query) 
    2222                self.cursor.execute(query) 
    23                 last = CommonDAO.lastID(self, self.jointable) 
     23                last = self.lastID(self.jointable) 
    2424                return last  
    2525        except Exception, err: 
  • donkeysurvey/database/FileDAO.py

    r152 r154  
    1818            logging.debug(query) 
    1919            self.cursor.execute(query) 
    20             last = CommonDAO.lastID(self, self.tablename) 
     20            last = self.lastID(self.tablename) 
    2121            return last 
    2222        except Exception, err: 
     
    3636                logging.debug(queryInsert) 
    3737                self.cursor.execute(queryInsert) 
    38                 last = CommonDAO.lastID(self, self.tablename) 
     38                last = self.lastID(self, self.tablename) 
    3939                return last 
    4040        except Exception, err: 
  • donkeysurvey/database/FileHasFilenameDAO.py

    r151 r154  
    1919            logging.debug(query) 
    2020            self.cursor.execute(query) 
    21         #CommonDAO.lastID(self, self.jointable) 
     21        #self.lastID(self.jointable) 
    2222 
    2323    def delete(self, fileId, filenameId): 
  • donkeysurvey/database/FilenameDAO.py

    r146 r154  
    1818            logging.debug(query) 
    1919            self.cursor.execute(query) 
    20             last = CommonDAO.lastID(self, self.tablename) 
     20            last = self.lastID(self.tablename) 
    2121            return last 
    2222        except Exception, err: 
     
    3131                logging.debug(query) 
    3232                self.cursor.execute(query) 
    33                 last = CommonDAO.lastID(self, self.tablename) 
     33                last = self.lastID(self.tablename) 
    3434                return last 
    3535            else: 
  • donkeysurvey/database/SessionDAO.py

    r151 r154  
    2020            (self.tablename, startDate, startDate, session.downloaded, \ 
    2121            session.uploaded, session.source.id, session.file.id, session.address.id) 
    22         logging.debug(query); 
    2322        try: 
     23            logging.debug(query); 
    2424            self.cursor.execute(query) 
     25            session.id = self.lastID(self.tablename) 
     26            return session.id  
    2527        except Exception, err: 
    2628            sys.stderr.write('ERROR: %s\n' % std(err)) 
    2729            return None 
    28         session.id = CommonDAO.lastID(self, self.tablename) 
    29         return session.id  
    3030 
    3131    def insertOrUpdate(self, session): 
     
    4747                logging.debug(queryInsert) 
    4848                self.cursor.execute(queryInsert) 
    49                 last = CommonDAO.lastID(self, self.tablename) 
     49                last = self.lastID(self.tablename) 
    5050                return last 
    5151        except Exception, err: 
  • donkeysurvey/database/SourceDAO.py

    r116 r154  
    1414 
    1515    def insert(self, source): 
    16         query = "INSERT INTO %s (name, hash, software, osinfo) values('%s', '%s', '%s', '%s')" % (self.tablename, source.name, source.hash, source.software, source.osinfo) 
     16        try: 
     17            query = "INSERT INTO %s (name, hash, software, osinfo) values('%s', '%s', '%s', '%s')" % (self.tablename, source.name, source.hash, source.software, source.osinfo) 
     18            logging.debug(query) 
     19            self.cursor.execute(query) 
     20            last = self.lastID(self.tablename) 
     21            return last 
     22        except Exception, err: 
     23            sys.stderr.write('ERROR: %s\n' % str(err)) 
     24            return None   
     25 
     26    def delete(self, id): 
     27        query = "DELETE FROM %s WHERE id = %s" % (self.tablename, id) 
    1728        logging.debug(query) 
    1829        self.cursor.execute(query) 
    19         last = self.lastID(self.tablename) 
    20         return last 
    21  
    22     def delete(self, id): 
    23         self.cursor.execute("""DELETE FROM """+self.tablename+""" WHERE id = %s""", (id,)) 
    2430 
    2531    def find(self, id): 
    2632        query = "SELECT * FROM %s WHERE id = %s" % (self.tablename, id) 
     33        logging.debug(query) 
    2734        self.cursor.execute(query) 
    2835        rs = self.cursor.fetchall() 
     
    4047    def findByHash(self, hash): 
    4148        query = "SELECT * FROM %s WHERE hash = '%s'" % (self.tablename, hash) 
     49        logging.debug(query) 
    4250        self.cursor.execute(query) 
    4351        rs = self.cursor.fetchall() 
  • donkeysurvey/database/SourceHasFileDAO.py

    r149 r154  
    1414     
    1515    def insertOrUpdate(self, sourceId, fileId, availability): 
    16         rs = self.findBySourceFile(sourceId, fileId) 
    17         if rs is not None: 
    18             queryUpdate = "UPDATE %s SET availability = %s WHERE source_id = %s and file_id = %s" % (self.jointable, availability, sourceId, fileId) 
    19             logging.debug(queryUpdate) 
    20             self.cursor.execute(queryUpdate)  
    21         else: 
    22             firstSeen = datetime.now().strftime("%Y-%m-%d %H:%M:%S") 
    23             queryInsert = "INSERT INTO %s(source_id, file_id, first_seen, availability) VALUES(%s, %s, '%s', %s)" % (self.jointable, sourceId, fileId, firstSeen, availability) 
    24             logging.debug(queryInsert) 
    25             self.cursor.execute(queryInsert) 
    26         #CommonDAO.lastID(self, self.jointable) 
     16        try: 
     17            rs = self.findBySourceFile(sourceId, fileId) 
     18            if rs is not None: 
     19                queryUpdate = "UPDATE %s SET availability = %s WHERE source_id = %s and file_id = %s" % (self.jointable, availability, sourceId, fileId) 
     20                logging.debug(queryUpdate) 
     21                self.cursor.execute(queryUpdate)  
     22            else: 
     23                firstSeen = datetime.now().strftime("%Y-%m-%d %H:%M:%S") 
     24                queryInsert = "INSERT INTO %s(source_id, file_id, first_seen, availability) VALUES(%s, %s, '%s', %s)" % (self.jointable, sourceId, fileId, firstSeen, availability) 
     25                logging.debug(queryInsert) 
     26                self.cursor.execute(queryInsert) 
     27        except Exception, err: 
     28            sys.stderr.write('ERROR: %s\n' % str(err)) 
     29            return None   
     30         
     31        #self.lastID(self.jointable) 
    2732 
    2833    def delete(self, sourceId, fileId): 
  • donkeysurvey/database/tests/AddressTestCase.py

    r137 r154  
    2121        assert aid != None, 'error inserting address' 
    2222         
    23   
    2423    def testSelect(self): 
    2524        adao = AddressDAO() 
  • donkeysurvey/database/tests/run.py

    r146 r154  
    33python FileTestCase.py  
    44python SessionTestCase.py 
    5 python Address.py 
     5python AddressTestCase.py