Use context manager for mssql and postgresql modules. (#65224)
This commit is contained in:
parent
deb0cbbf73
commit
d335d7a62c
2 changed files with 12 additions and 16 deletions
|
@ -122,8 +122,7 @@ def db_delete(conn, cursor, db):
|
|||
|
||||
def db_import(conn, cursor, module, db, target):
|
||||
if os.path.isfile(target):
|
||||
backup = open(target, 'r')
|
||||
try:
|
||||
with open(target, 'r') as backup:
|
||||
sqlQuery = "USE [%s]\n" % db
|
||||
for line in backup:
|
||||
if line is None:
|
||||
|
@ -135,8 +134,6 @@ def db_import(conn, cursor, module, db, target):
|
|||
sqlQuery += line
|
||||
cursor.execute(sqlQuery)
|
||||
conn.commit()
|
||||
finally:
|
||||
backup.close()
|
||||
return 0, "import successful", ""
|
||||
else:
|
||||
return 1, "cannot find target file", "cannot find target file"
|
||||
|
|
|
@ -285,7 +285,7 @@ class PgHba(object):
|
|||
self.comment = []
|
||||
# read the pg_hbafile
|
||||
try:
|
||||
file = open(self.pg_hba_file, 'r')
|
||||
with open(self.pg_hba_file, 'r') as file:
|
||||
for line in file:
|
||||
line = line.strip()
|
||||
# uncomment
|
||||
|
@ -296,7 +296,6 @@ class PgHba(object):
|
|||
self.add_rule(PgHbaRule(line=line))
|
||||
except PgHbaRuleError:
|
||||
pass
|
||||
file.close()
|
||||
self.unchanged()
|
||||
except IOError:
|
||||
pass
|
||||
|
|
Loading…
Reference in a new issue