strausTransf

import os
import xml.etree.ElementTree as ET


alert_file = None


def set_double_value_for(target_field):
    current_precision_string = target_field.get('PRECISION')
    print('\n' + current_precision_string)
    double_precision_value = int(current_precision_string) * 2
    again_as_string_value = str(double_precision_value)
    target_field.set('PRECISION', again_as_string_value)


def create_alert_file_if_date_is_string(target_field, original_file_name):
    if target_field.get('NAME').startswith('DT_'):
        alert_file = open('../zoutput/{0}_b_alert.txt'.format(original_file_name[:-4]), 'a')  # supress -.xml
        alert_file.write('\n')
        alert_file.write(target_field.get('NAME') + ' (string)')
        alert_file.write('\n')
        alert_file.write(str(target_field.attrib))
        alert_file.write('\n')


all_original_files = os.listdir('../original')
for each_original_file in all_original_files:
    tree = ET.parse('../original/'+each_original_file)
    root = tree.getroot()
    for target_field in root.iter('TARGETFIELD'):
        if target_field.get('DATATYPE') in ('string', 'nstring'):
            set_double_value_for(target_field)
            create_alert_file_if_date_is_string(target_field, each_original_file)
    tree.write('../zoutput/{0}_a_output.xml'.format(each_original_file[:-4]))                # supress -.xml
    if alert_file:
        alert_file.close()

#