def setUp(self):
super().setUp()
self.validator = Validator()
self.validator.flush_names()
logging.getLogger().setLevel(logging.CRITICAL)
FileUtil.safe_delete_dir('in')
FileUtil.safe_delete_dir('out')
FileUtil.safe_create_dir('in')
FileUtil.safe_create_dir('out')
cols = dict()
cols['name'] = 'National 2m'
cols['medium_name'] = 'Natl 2m'
cols['short_name'] = 'NATL 2M'
cols['zone_id'] = ''
cols['rx_freq'] = '146.52'
cols['rx_ctcss'] = ''
cols['rx_dcs'] = ''
cols['rx_dcs_invert'] = ''
cols['tx_power'] = 'High'
cols['tx_offset'] = ''
cols['tx_ctcss'] = ''
cols['tx_dcs'] = ''
cols['tx_dcs_invert'] = ''
cols['digital_timeslot'] = ''
cols['digital_color'] = ''
cols['digital_contact_id'] = ''
cols['latitude'] = ''
cols['longitude'] = ''
self.radio_cols = cols
def test_migration_one(self):
FileUtil.safe_delete_dir('in')
FileUtil.safe_delete_dir('out')
self.manager._migrate_one()
self.assertTrue(os.path.exists('in'))
self.assertTrue(os.path.exists('out'))
def setUp(self):
FileUtil.safe_delete_dir('in')
FileUtil.safe_delete_dir('out')
FileUtil.safe_create_dir('in')
FileUtil.safe_create_dir('out')
PathManager.set_input_path('./in')
PathManager.set_output_path('./out')
def test_sets_different_path(self):
os.mkdir('./whoa_different')
PathManager.set_input_path('whoa_different')
w = PathManager.open_input_file('x.tst', 'w+')
w.write('different test success')
w.close()
f = open('./whoa_different/x.tst', 'r')
line = f.readline()
f.close()
self.assertEqual(line, 'different test success')
FileUtil.safe_delete_dir('./whoa_different')
def setUp(self):
super().setUp()
logging.getLogger().setLevel(logging.ERROR)
FileUtil.safe_delete_dir('./in')
FileUtil.safe_delete_dir('./out')
FileUtil.safe_delete_dir('./whoa_different')
os.mkdir('./in')
os.mkdir('./out')
self.manager = MigrationManager()
def cleanup(self):
FileUtil.safe_delete_dir(PathManager.get_input_path())
FileUtil.safe_delete_dir(PathManager.get_output_path())
def generate_all_declared(self):
file_errors = self._validator.validate_files_exist()
self._validator.flush_names()
if len(file_errors) > 0:
return False
results = self._migrations.check_migrations_needed()
if len(results.keys()) > 0:
logging.warning(
'You may be using an old version of the input files. Have you run migrations?'
)
logging.warning("Migrations check is under the 'File' menu.")
sleep(1)
for radio in radio_types.radio_choices():
radio_folder = PathManager.get_output_path(radio)
if not os.path.exists(radio_folder):
continue
logging.info(f'Deleting old output folder `{radio}`')
FileUtil.safe_delete_dir(radio_folder)
digital_contacts, digi_contact_errors = self._generate_digital_contact_data(
)
dmr_ids, dmr_id_errors = self._generate_dmr_id_data()
zones, zone_errors = self._generate_zone_data()
user, user_data_errors = self._generate_user_data()
preload_errors = digi_contact_errors + dmr_id_errors + zone_errors + user_data_errors
feed = PathManager.open_input_file('input.csv', 'r')
csv_reader = csv.DictReader(feed)
radio_channel_errors = []
radio_channels = []
line_num = 1
for line in csv_reader:
line_errors = self._validator.validate_radio_channel(
line, line_num, feed.name, digital_contacts, zones)
radio_channel_errors += line_errors
line_num += 1
if len(line_errors) > 0:
continue
radio_channel = RadioChannel(line, digital_contacts, dmr_ids)
radio_channels.append(radio_channel)
if radio_channel.zone_id.fmt_val(None) is not None:
zones[radio_channel.zone_id.fmt_val()].add_channel(
radio_channel)
feed.close()
all_errors = preload_errors + radio_channel_errors
if len(all_errors) > 0:
logging.error('--- VALIDATION ERRORS, CANNOT CONTINUE ---')
for err in all_errors:
logging.error(
f'\t\tfile: `{err.file_name}` line:{err.line_num} validation error: {err.message}'
)
return False
else:
logging.info(
'File validation complete, no obvious formatting errors found')
radio_files = dict()
headers_gen = RadioChannel.create_empty()
FileUtil.safe_create_dir('out')
channel_numbers = dict()
for radio in self.radio_list:
radio_casted = RadioChannelBuilder.casted(headers_gen, radio)
FileUtil.safe_create_dir(f'out/{radio}')
logging.info(f'Generating for radio type `{radio}`')
if radio_casted.skip_radio_csv():
logging.info(
f'`{radio}` uses special output style. Skipping channels csv'
)
continue
output = RadioWriter.output_writer(f'{radio}/{radio}_channels.csv',
'\r\n')
file_headers = radio_casted.headers()
output.writerow(file_headers)
radio_files[radio] = output
channel_numbers[radio] = 1
logging.info('Processing radio channels')
line = 1
for radio_channel in radio_channels:
logging.debug(f'Processing radio line {line}')
if line % file_util.RADIO_LINE_LOG_INTERVAL == 0:
logging.info(f'Processing radio line {line}')
line += 1
for radio in self.radio_list:
if radio not in radio_files.keys():
continue
if not radio_types.supports_dmr(
radio) and radio_channel.is_digital():
continue
casted_channel = RadioChannelBuilder.casted(
radio_channel, radio)
input_data = casted_channel.output(channel_numbers[radio])
radio_files[radio].writerow(input_data)
channel_numbers[radio] += 1
additional_data = RadioAdditional(radio_channels, dmr_ids,
digital_contacts, zones, user)
for radio in self.radio_list:
if radio in radio_files.keys():
radio_files[radio].close()
casted_additional_data = RadioAdditionalBuilder.casted(
additional_data, radio)
casted_additional_data.output()
logging.info(f'''Radio generator complete. Your output files are in
`{os.path.abspath('out')}`
The next step is to import these files into your radio programming application. (e.g. CHiRP)'''
)
return True
def setUp(self):
super().setUp()
logging.getLogger().setLevel(logging.ERROR)
FileUtil.safe_delete_dir('in')
FileUtil.safe_delete_dir('out')
self.manager = MigrationManager()
def setUp(self):
super().setUp()
FileUtil.safe_delete_dir('in')
FileUtil.safe_delete_dir('out')
self.wizard = Wizard()
def test_delete_no_exception(self):
os.mkdir('del_test')
self.assertTrue(os.path.exists('del_test'))
FileUtil.safe_delete_dir('del_test')
FileUtil.safe_delete_dir('del_test')
self.assertFalse(os.path.exists('del_test'))