Don't hardcode exit values, use os.sync() in ddusb

This commit is contained in:
Eric Torres 2019-03-16 22:19:18 -07:00
parent 5660ef32ad
commit e165d6768f

View File

@ -2,6 +2,7 @@
"""Wrapper script for using dd to write to a USB drive."""
import argparse
import os
import pathlib
import re
import subprocess
@ -10,6 +11,10 @@ import subprocess
COMMENT_PATTERN = "^[#;]"
EXCLUDE_FILE = "/etc/helper-scripts/ddusb-exclude.conf"
E_BLOCKDEVICE_ERROR = 1
E_EXCLUDE_ERROR = 2
E_DD_ERROR = 3
# ========== Functions ==========
def read_exclude_file(exclude_file):
@ -46,7 +51,7 @@ block_path = args.output_file
# Ensure that block_path is really a block device
if not pathlib.Path(block_path).is_block_device():
print(f'Error: "{block_path}" is not a block device')
exit(1)
exit(E_BLOCKDEVICE_ERROR)
# Check if block_path is excluded
exclude_patterns = read_exclude_file(EXCLUDE_FILE)
@ -54,7 +59,7 @@ exclude_patterns = read_exclude_file(EXCLUDE_FILE)
for pattern in exclude_patterns:
if re.fullmatch(pattern, block_path):
print(f'Error: "{block_path}" is blacklisted from running dd')
exit(2)
exit(E_EXCLUDE_ERROR)
print(f"Input file: {input_file}")
print(f"Block device: {block_path}")
@ -72,6 +77,6 @@ try:
check=True,
)
except subprocess.CalledProcessError:
exit(3)
exit(E_DD_ERROR)
else:
subprocess.run("sync")
os.sync()