2012-07-24 64 views
0

我已经使用sqlalchemy(我的底层数据库是PostgreSQL 8.4)编写了以下函数。 我有以下模式:sqlalchemy错误地删除行

CREATE TABLE temporary_data_key (
         id       SERIAL PRIMARY KEY, 
         key       VARCHAR(36) NOT NULL 
         ); 
CREATE UNIQUE INDEX idxu_temp_data_nm ON temporary_data_key (key); 


CREATE TABLE temporary_data (
         key_id      INTEGER REFERENCES temporary_data_key(id) ON UPDATE CASCADE ON DELETE RESTRICT NOT NULL, 
         id_value     INTEGER CHECK (id_value > 0) NOT NULL 
         ); 

我期待的功能,清理后本身(即删除temporary_table_key表生成的密钥),如果出现错误 - 由于某种原因,该表中没有被清除时发生异常。

下面是函数:

def storeIdsInTemporaryTable(dbinfo, id_list): 
    conn = dbinfo['db_connection'] 
    metadata = dbinfo['metadata'] 
    tableinfo = Table('temporary_data', metadata, autoload=True) 
    datarows = [] 

    temp_dbinfo = {'db_connection': conn, 'table': tableinfo, 'datarows': datarows } 

    guid = genutils.getGUID() 

    sql = "INSERT INTO temporary_data_key (key) VALUES ('{0}') RETURNING id".format(guid) 
    key_id = executeSQLonDbConnection(temp_dbinfo, sql, return_field='id') 

    for id_value in id_list: 
     datarows.append({ 'key_id': key_id, 'id_value': id_value }) 

    try: 
     insertToDb(temp_dbinfo) 
    except Exception as e: 
     print 'ERROR:',e 
     guid = None # to indicate an error occured 
     if key_id: 
      conn.execute("DELETE FROM temporary_data_key WHERE key='{0}'".format(key_id)) 

    return guid 

任何人都可以找出为什么功能不清理temporary_data密钥,作为例外处理部分?

回答

1

我找到了解决方案。

最后一行应该是:

if key_id: 
    conn.execute("DELETE FROM temporary_data_key WHERE key='{0}'".format(guid))