import MigrationCommon as MC
import CheckoutCommon as CC
import Checkout4 as C4


def count_r2d2_lines(state_of_interest):
    pp_workspace = MC.get_r2d2_preprod_workspace(state_of_interest)
    pp_lines = CC.list_lines_fcs(pp_workspace)
    pp_dict = CC.count_list_fcs(pp_lines, C4.count_r2d2_fc_state)

    return pp_dict


#
# start of program
#
if __name__ == "__main__":
    log_name = 'checkout5'
    logger = MC.setup_logging(log_name)
    logger.info("Starting program...")

    CC.write_rd2d2_by_state(log_name, logger, count_r2d2_lines)

    logger.info("Finished program")
import arcpy
import MigrationCommon as MC
import CheckoutCommon as CC
import Checkout4 as C4


def count_r2d2_wap(state_of_interest):
    p_workspace = MC.get_r2d2_prod_workspace(state_of_interest)
    arcpy.env.workspace = p_workspace
    p_wap = ['NISOURCE.GPSWorkAreaPolygon']
    p_dict = CC.count_list_fcs(p_wap, C4.count_r2d2_fc_state)

    return p_dict


#
# start of program
#
if __name__ == "__main__":
    log_name = 'checkout6'
    logger = MC.setup_logging(log_name)
    logger.info("Starting program...")

    CC.write_rd2d2_by_state(log_name, logger, count_r2d2_wap)

    logger.info("Finished program")