def test_create_stanza(self):
service = Service([
Stanza('script://test1', {}),
Stanza('package://test2', {}),
Stanza('script://test3', {}),
])
r_config.create_stanza(service, 'test', 'test4', {'a': 'b'})
cnt = 0
for stanza, name in r_config.iter_stanzas(service, 'test'):
self.assertEqual(cnt, 0)
cnt += 1
self.assertEqual(getattr(stanza, 'a'), 'b')
def test_create_stanza(self):
service = Service([
Stanza('script://test1', {}),
Stanza('package://test2', {}),
Stanza('script://test3', {}),
])
r_config.create_stanza(service, 'test', 'test4', {
'a': 'b'
})
cnt = 0
for stanza, name in r_config.iter_stanzas(service, 'test'):
self.assertEqual(cnt, 0)
cnt += 1
self.assertEqual(getattr(stanza, 'a'), 'b')
def create_files(service):
lock_file_path = r_path.get_file('scripts.lock')
with r_lockfile.file_lock(lock_file_path):
custom_scripts_path = get_custom_scripts_path()
filenames = set()
# check configuration for requred scripts
for stanza, name in r_config.iter_stanzas(service, scheme):
filename = name + os.path.extsep + extension
full_path = os.path.join(custom_scripts_path, filename)
# in case there is no uploaded attribute, add it with the current timestamp
if not hasattr(stanza, uploaded_stanza_key):
stanza = r_config.create_stanza(
service, scheme, name, {
'content': stanza.__getattr__('content'),
uploaded_stanza_key: int(time.time())
})
if os.path.exists(full_path):
# compare file creation date with stanza creation date
modify_time = int(os.path.getmtime(full_path))
uploaded_time = int(getattr(stanza, uploaded_stanza_key))
if uploaded_time > modify_time:
os.remove(full_path)
create_file = True
else:
create_file = False
else:
create_file = True
if create_file:
script_content = base64.decodestring(
stanza.__getattr__('content'))
with open(full_path, 'wb') as f:
f.write(script_content)
filenames.add(filename)
# remove files that are no longer in the list of configures scripts
for filename in os.listdir(custom_scripts_path):
if filename.endswith(os.path.extsep + extension):
if not filename in filenames:
script_path = os.path.join(custom_scripts_path, filename)
os.remove(script_path)
def create_files(service):
lock_file_path = r_path.get_file('scripts.lock')
with r_lockfile.file_lock(lock_file_path):
custom_scripts_path = get_custom_scripts_path()
filenames = set()
# check configuration for requred scripts
for stanza, name in r_config.iter_stanzas(service, scheme):
filename = name + os.path.extsep + extension
full_path = os.path.join(custom_scripts_path, filename)
# in case there is no uploaded attribute, add it with the current timestamp
if not hasattr(stanza, uploaded_stanza_key):
stanza = r_config.create_stanza(service, scheme, name, {
'content': stanza.__getattr__('content'),
uploaded_stanza_key: int(time.time())
})
if os.path.exists(full_path):
# compare file creation date with stanza creation date
modify_time = int(os.path.getmtime(full_path))
uploaded_time = int(getattr(stanza, uploaded_stanza_key))
if uploaded_time > modify_time:
os.remove(full_path)
create_file = True
else:
create_file = False
else:
create_file = True
if create_file:
script_content = base64.decodestring(stanza.__getattr__('content'))
with open(full_path, 'wb') as f:
f.write(script_content)
filenames.add(filename)
# remove files that are no longer in the list of configures scripts
for filename in os.listdir(custom_scripts_path):
if filename.endswith(os.path.extsep + extension):
if not filename in filenames:
script_path = os.path.join(custom_scripts_path, filename)
os.remove(script_path)
def add(service, name):
r_config.create_stanza(service, scheme, name, {
})
def add(service, name, content):
r_config.create_stanza(service, scheme, name, {
'content': base64.encodestring(content).replace('\n', ''),
uploaded_stanza_key: int(time.time())
})
def add(service, name, content):
r_config.create_stanza(
service, scheme, name, {
'content': base64.encodestring(content).replace('\n', ''),
uploaded_stanza_key: int(time.time())
})