def test_create_new_existing(self):
"""Test connecting to the instance database if it exists."""
db_file = tempfile.NamedTemporaryFile()
db_location = 'sqlite:///{}'.format(db_file.name)
db.create(db_location)
# Test creating if already exists
engine = db.create(db_location)
meta = MetaData(bind=engine)
meta.reflect(bind=engine)
assert 'patient' in meta.tables
assert 'study' in meta.tables
assert 'series' in meta.tables
assert 'image' in meta.tables
assert 'instance' in meta.tables
def setup(self):
"""Run prior to each test"""
engine = db.create('sqlite:///:memory:')
pydicom.config.use_none_as_empty_text_VR_value = True
self.session = sessionmaker(bind=engine)()
for fname in DATASETS:
fpath = os.path.join(DATA_DIR, fname)
ds = dcmread(fpath)
db.add_instance(ds, self.session)
def setup(self):
"""Run prior to each test"""
engine = db.create('sqlite:///:memory:')
pydicom.config.use_none_as_empty_text_VR_value = True
ds = Dataset()
ds.PatientID = '1234'
ds.StudyInstanceUID = '1.2'
ds.SeriesInstanceUID = '1.2.3'
ds.SOPInstanceUID = '1.2.3.4'
self.minimal = ds
self.session = sessionmaker(bind=engine)()
def test_create_new(self):
"""Test connecting to the instance database if it doesn't exist."""
db_file = tempfile.NamedTemporaryFile()
db_location = f'sqlite:///{db_file.name}'
engine = db.create(db_location)
# Check exists with tables
meta = MetaData(bind=engine)
meta.reflect(bind=engine)
assert 'patient' in meta.tables
assert 'study' in meta.tables
assert 'series' in meta.tables
assert 'image' in meta.tables
assert 'instance' in meta.tables
def setup(self):
"""Run prior to each test"""
engine = db.create("sqlite:///:memory:")
pydicom.config.use_none_as_empty_text_VR_value = True
self.session = sessionmaker(bind=engine)()
ds = Dataset()
ds.PatientID = "1234"
ds.StudyInstanceUID = "1.2"
ds.SeriesInstanceUID = "1.2.3"
ds.SOPInstanceUID = "1.2.3.4"
db.add_instance(ds, self.session)
self.minimal = ds
def main(args=None):
"""Run the application."""
if args is not None:
sys.argv = args
args = _setup_argparser()
if args.version:
print(f"qrscp.py v{__version__}")
sys.exit()
APP_LOGGER = setup_logging(args, "qrscp")
APP_LOGGER.debug(f"qrscp.py v{__version__}")
APP_LOGGER.debug("")
APP_LOGGER.debug("Using configuration from:")
APP_LOGGER.debug(f" {args.config}")
APP_LOGGER.debug("")
config = ConfigParser()
config.read(args.config)
if args.ae_title:
config["DEFAULT"]["ae_title"] = args.ae_title
if args.port:
config["DEFAULT"]["port"] = args.port
if args.max_pdu:
config["DEFAULT"]["max_pdu"] = args.max_pdu
if args.acse_timeout:
config["DEFAULT"]["acse_timeout"] = args.acse_timeout
if args.dimse_timeout:
config["DEFAULT"]["dimse_timeout"] = args.dimse_timeout
if args.network_timeout:
config["DEFAULT"]["network_timeout"] = args.network_timeout
if args.bind_address:
config["DEFAULT"]["bind_address"] = args.bind_address
if args.database_location:
config["DEFAULT"]["database_location"] = args.database_location
if args.instance_location:
config["DEFAULT"]["instance_location"] = args.instance_location
# Log configuration settings
_log_config(config, APP_LOGGER)
app_config = config["DEFAULT"]
dests = {}
for ae_title in config.sections():
dest = config[ae_title]
# Convert to bytes and validate the AE title
ae_title = set_ae(ae_title, "ae_title", False, False)
dests[ae_title] = (dest["address"], dest.getint("port"))
# Use default or specified configuration file
current_dir = os.path.abspath(os.path.dirname(__file__))
instance_dir = os.path.join(current_dir, app_config["instance_location"])
db_path = os.path.join(current_dir, app_config["database_location"])
# The path to the database
db_path = f"sqlite:///{db_path}"
db.create(db_path)
# Clean up the database and storage directory
if args.clean:
response = input(
"This will delete all instances from both the storage directory "
"and the database. Are you sure you wish to continue? [yes/no]: ")
if response != "yes":
sys.exit()
if clean(db_path, instance_dir, APP_LOGGER):
sys.exit()
else:
sys.exit(1)
# Try to create the instance storage directory
os.makedirs(instance_dir, exist_ok=True)
ae = AE(app_config["ae_title"])
ae.maximum_pdu_size = app_config.getint("max_pdu")
ae.acse_timeout = app_config.getfloat("acse_timeout")
ae.dimse_timeout = app_config.getfloat("dimse_timeout")
ae.network_timeout = app_config.getfloat("network_timeout")
## Add supported presentation contexts
# Verification SCP
ae.add_supported_context(Verification, ALL_TRANSFER_SYNTAXES)
# Storage SCP - support all transfer syntaxes
for cx in AllStoragePresentationContexts:
ae.add_supported_context(cx.abstract_syntax,
ALL_TRANSFER_SYNTAXES,
scp_role=True,
scu_role=False)
# Query/Retrieve SCP
ae.add_supported_context(PatientRootQueryRetrieveInformationModelFind)
ae.add_supported_context(PatientRootQueryRetrieveInformationModelMove)
ae.add_supported_context(PatientRootQueryRetrieveInformationModelGet)
ae.add_supported_context(StudyRootQueryRetrieveInformationModelFind)
ae.add_supported_context(StudyRootQueryRetrieveInformationModelMove)
ae.add_supported_context(StudyRootQueryRetrieveInformationModelGet)
# Set our handler bindings
handlers = [
(evt.EVT_C_ECHO, handle_echo, [args, APP_LOGGER]),
(evt.EVT_C_FIND, handle_find, [db_path, args, APP_LOGGER]),
(evt.EVT_C_GET, handle_get, [db_path, args, APP_LOGGER]),
(evt.EVT_C_MOVE, handle_move, [dests, db_path, args, APP_LOGGER]),
(evt.EVT_C_STORE, handle_store,
[instance_dir, db_path, args, APP_LOGGER]),
]
# Listen for incoming association requests
ae.start_server((app_config["bind_address"], app_config.getint("port")),
evt_handlers=handlers)
