gaming/lutris
gaming/lutris
1
0
Fork 0
mirror of https://github.com/lutris/lutris synced 2024-10-06 15:59:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add metas to registry keys

Browse source
This commit is contained in:
Mathieu Comandon 2016-10-25 12:54:47 -07:00
parent 6cb25e619f
commit 01e3c5f637
2 changed files with 23 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lutris/util/wineregistry.py
View file

@ -67,6 +67,8 @@ class WineRegistry(object):
current_key = None
key_index = 0
for line in registry_lines:
# Remove trailing newlines
line = line.rstrip('\n')
if line.startswith('#arch'):
self.arch = line.split('=')[1]
continue
@ -77,7 +79,9 @@ class WineRegistry(object):
key_index += 1
continue
if current_key:
if line.startswith('"'):
if line.startswith('#'):
current_key.add_meta(line)
elif line.startswith('"'):
k, v = line.split('=', 1)
current_key.set_key(k, v)
elif line.startswith('@'):
@ -114,13 +118,17 @@ class WineRegistry(object):
class WineRegistryKey(object):
def __init__(self, key_def):
self.raw_name = key_def[:key_def.index(']') + 1]
# Parse timestamp either as int or float
self.raw_timestamp = key_def[key_def.index(']') + 1:]
ts_parts = self.raw_timestamp.strip().split()
if len(ts_parts) == 1:
self.timestamp = int(ts_parts[0])
else:
self.timestamp = float("{}.{}".format(ts_parts[0], ts_parts[1]))
self.values = {}
self.metas = {}
self.name = self.raw_name.replace('\\\\', '/').strip("[]")
def set_key(self, name, value):
@ -129,6 +137,16 @@ class WineRegistryKey(object):
def __str__(self):
return "{0} {1}".format(self.raw_name, self.raw_timestamp)
def add_meta(self, meta_line):
if not meta_line.startswith('#'):
raise ValueError("Key metas should start with '#'")
meta_line = meta_line[1:]
key, value = meta_line.split('=')
self.metas[key] = value
def get_meta(self, name):
return self.metas.get(name)
def get_value(self, name):
if name not in self.values:
return

4
tests/test_registry.py
View file

@ -24,3 +24,7 @@ class TestWineRegistry(TestCase):
def test_can_get_timestamp_as_float(self):
key = self.registry.get_key('Control Panel/Sound')
self.assertEqual(key.timestamp, 1475423303.7943190)
def test_can_get_meta(self):
key = self.registry.get_key('Control Panel/Sound')
self.assertEqual(key.get_meta('time'), '1d21cc468677196')