benya/whipper-gui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

config: generalize getting and setting of drive options

Browse Source 
Fixed merge conflicts (JoeLametta).

Signed-off-by: Andreas Oberritter <obi@saftware.de>
This commit is contained in:
Andreas Oberritter
2018-05-09 14:01:47 +02:00
committed by JoeLametta
parent 00b0692142
commit afc31f930e
1 changed files with 26 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
whipper/common/config.py
View File

@@ -86,43 +86,20 @@ class Config:
# drive sections
def setReadOffset(self, vendor, model, release, offset):
"""
Set a read offset for the given drive.
Strips the given strings of leading and trailing whitespace.
"""
section = self._findOrCreateDriveSection(vendor, model, release)
self._parser.set(section, 'read_offset', str(offset))
self.write()
"""Set a read offset for the given drive."""
self._setDriveOption(vendor, model, release, 'read_offset', offset)
def getReadOffset(self, vendor, model, release):
"""Get a read offset for the given drive."""
section = self._findDriveSection(vendor, model, release)
try:
return int(self._parser.get(section, 'read_offset'))
except configparser.NoOptionError:
raise KeyError("Could not find read_offset for %s/%s/%s" % (
vendor, model, release))
return int(self._getDriveOption(vendor, model, release, 'read_offset'))
def setDefeatsCache(self, vendor, model, release, defeat):
"""
Set whether the drive defeats the cache.
Strips the given strings of leading and trailing whitespace.
"""
section = self._findOrCreateDriveSection(vendor, model, release)
self._parser.set(section, 'defeats_cache', str(defeat))
self.write()
"""Set whether the drive defeats the cache."""
self._setDriveOption(vendor, model, release, 'defeats_cache', defeat)
def getDefeatsCache(self, vendor, model, release):
section = self._findDriveSection(vendor, model, release)
try:
return self._parser.get(section, 'defeats_cache') == 'True'
except configparser.NoOptionError:
raise KeyError("Could not find defeats_cache for %s/%s/%s" % (
vendor, model, release))
option = self._getDriveOption(vendor, model, release, 'defeats_cache')
return option == 'True'
def _findDriveSection(self, vendor, model, release):
for name in self._parser.sections():
@@ -161,3 +138,22 @@ class Config:
self.write()
return self._findDriveSection(vendor, model, release)
def _getDriveOption(self, vendor, model, release, key):
"""Get an option for the given drive."""
section = self._findDriveSection(vendor, model, release)
try:
return self._parser.get(section, key)
except configparser.NoOptionError:
raise KeyError("Could not find %s for %s/%s/%s" % (
key, vendor, model, release))
def _setDriveOption(self, vendor, model, release, key, value):
"""
Set an option for the given drive.
Strips the given strings of leading and trailing whitespace.
"""
section = self._findOrCreateDriveSection(vendor, model, release)
self._parser.set(section, key, str(value))
self.write()