PsychoticNinja/JLU5-SupyPlugins
1
0
Fork 0
mirror of https://github.com/jlu5/SupyPlugins.git synced 2025-05-02 16:31:11 -05:00
Code Issues Projects Releases Wiki Activity

accountsdb: write DB to a temporary file first

Browse Source
This commit is contained in:
James Lu 2019-03-09 11:15:34 -08:00
parent f3bfd9ff73
commit e46915b58d
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
LastFM/local/accountsdb.py
View File

@ -86,9 +86,11 @@ class AccountsDB():
def flush(self): def flush(self):
"""Exports the database to a file.""" """Exports the database to a file."""
tmp_filename = self.filename + '.tmp'
try: try:
with open(self.filename, 'wb') as f: with open(tmp_filename, 'wb') as f:
pickle.dump(self.db, f, 2) pickle.dump(self.db, f, 2)
os.rename(tmp_filename, self.filename)
except Exception as e: except Exception as e:
log.warning('%s: Unable to write database: %s', self._plugin_name, e) log.warning('%s: Unable to write database: %s', self._plugin_name, e)

4
NuWeather/local/accountsdb.py
View File

@ -86,9 +86,11 @@ class AccountsDB():
def flush(self): def flush(self):
"""Exports the database to a file.""" """Exports the database to a file."""
tmp_filename = self.filename + '.tmp'
try: try:
with open(self.filename, 'wb') as f: with open(tmp_filename, 'wb') as f:
pickle.dump(self.db, f, 2) pickle.dump(self.db, f, 2)
os.rename(tmp_filename, self.filename)
except Exception as e: except Exception as e:
log.warning('%s: Unable to write database: %s', self._plugin_name, e) log.warning('%s: Unable to write database: %s', self._plugin_name, e)

4
accountsdb.py
View File

@ -85,9 +85,11 @@ class AccountsDB():
def flush(self): def flush(self):
"""Exports the database to a file.""" """Exports the database to a file."""
tmp_filename = self.filename + '.tmp'
try: try:
with open(self.filename, 'wb') as f: with open(tmp_filename, 'wb') as f:
pickle.dump(self.db, f, 2) pickle.dump(self.db, f, 2)
os.rename(tmp_filename, self.filename)
except Exception as e: except Exception as e:
log.warning('%s: Unable to write database: %s', self._plugin_name, e) log.warning('%s: Unable to write database: %s', self._plugin_name, e)