#!/usr/bin/env python3
# SPDX-License-Identifier: LGPL-2.1-or-later
-OUTFILE_HEADER = """#!/usr/bin/env python3
+import filecmp
+import os
+import stat
+import subprocess
+import sys
+import tempfile
+
+OUTFILE_HEADER = '''#!/usr/bin/env python3
# SPDX-License-Identifier: LGPL-2.1-or-later
#
# create-sys-script.py
#
# © 2017 Canonical Ltd.
# Author: Dan Streetman <dan.streetman@canonical.com>
-"""
+'''
# Use this only to (re-)create the test/sys-script.py script,
# after adding or modifying anything in the test/sys/ directory
-import os, sys
-import stat
-import tempfile
-import filecmp
-import subprocess
-
OUTFILE_MODE = 0o775
-OUTFILE_FUNCS = r"""
+OUTFILE_FUNCS = r'''
import os, sys
import shutil
with open(path, "wb") as f:
f.write(contents)
os.chmod(path, mode)
-"""
+'''
-OUTFILE_MAIN = """
+OUTFILE_MAIN = '''
if len(sys.argv) < 2:
exit("Usage: {} <target dir>".format(sys.argv[0]))
if os.path.exists('sys'):
shutil.rmtree('sys')
-"""
+'''
def handle_dir(outfile, path):
r = repr(b)[2:-1]
# python escapes all ' only if there are ' and " in the string
if '"' not in r:
- r = r.replace("'", r"\'")
+ r = r.replace("'", r'\'')
# return line with all ' escaped
return r
def handle_file(outfile, path):
m = os.lstat(path).st_mode & 0o777
- with open(path, "rb") as f:
+ with open(path, 'rb') as f:
b = f.read()
- if b.count(b"\n") > 1:
- r = "\n".join( escape_single_quotes(l) for l in b.split(b"\n") )
+ if b.count(b'\n') > 1:
+ r = '\n'.join(escape_single_quotes(line) for line in b.split(b'\n'))
r = f"b'''{r}'''"
else:
r = repr(b)
def process_sysdir(outfile):
- for (dirpath, dirnames, filenames) in os.walk('sys'):
+ for dirpath, dirnames, filenames in os.walk('sys'):
handle_dir(outfile, dirpath)
for d in dirnames:
path = os.path.join(dirpath, d)
mode_a = os.lstat(path_a).st_mode
mode_b = os.lstat(path_b).st_mode
if not stat.S_ISDIR(mode_b):
- raise Exception("Not directory")
+ raise Exception('Not directory')
if (mode_a & 0o777) != (mode_b & 0o777):
- raise Exception("Permissions mismatch")
+ raise Exception('Permissions mismatch')
def verify_link(tmpd, path_a):
path_b = os.path.join(tmpd, path_a)
if not stat.S_ISLNK(os.lstat(path_b).st_mode):
- raise Exception("Not symlink")
+ raise Exception('Not symlink')
if os.readlink(path_a) != os.readlink(path_b):
- raise Exception("Symlink dest mismatch")
+ raise Exception('Symlink dest mismatch')
def verify_file(tmpd, path_a):
mode_a = os.lstat(path_a).st_mode
mode_b = os.lstat(path_b).st_mode
if not stat.S_ISREG(mode_b):
- raise Exception("Not file")
+ raise Exception('Not file')
if (mode_a & 0o777) != (mode_b & 0o777):
- raise Exception("Permissions mismatch")
+ raise Exception('Permissions mismatch')
if not filecmp.cmp(path_a, path_b, shallow=False):
- raise Exception("File contents mismatch")
+ raise Exception('File contents mismatch')
def verify_script(tmpd):
any = False
- for (dirpath, dirnames, filenames) in os.walk("sys"):
+ for dirpath, dirnames, filenames in os.walk('sys'):
any = True
try:
path = dirpath
if not any:
exit('Nothing found!')
-if __name__ == "__main__":
+
+if __name__ == '__main__':
if len(sys.argv) < 2:
exit('Usage: create-sys-script.py /path/to/test/')
os.chdir(sys.argv[1])
- with open(outfile, "w") as f:
+ with open(outfile, 'w') as f:
os.chmod(outfile, OUTFILE_MODE)
- f.write(OUTFILE_HEADER.replace(os.path.basename(sys.argv[0]),
- os.path.basename(outfile)))
+ f.write(OUTFILE_HEADER.replace(os.path.basename(sys.argv[0]), os.path.basename(outfile)))
f.write(OUTFILE_FUNCS)
f.write(OUTFILE_MAIN)
process_sysdir(f)