def test_get_transition_for_datetime(self):
zone_specifier = ZoneSpecifier(
zone_infos.ZONE_INFO_America_Los_Angeles)
# Just after a DST transition
dt = datetime(2000, 4, 2, 3, 0, 0)
transition = zone_specifier.get_transition_for_datetime(dt)
self.assertIsNotNone(transition)
# DST gap does not exist, but a transition should be returned.
dt = datetime(2000, 4, 2, 2, 59, 59)
transition = zone_specifier.get_transition_for_datetime(dt)
self.assertIsNotNone(transition)
def main() -> None:
# Configure command line flags.
parser = argparse.ArgumentParser(description='Zone Agent.')
parser.add_argument(
'--viewing_months',
help='Number of months to use for calculations (12, 13, 14, 36)',
type=int,
default=14)
parser.add_argument(
'--transition',
help='Print the transition instead of timezone info',
action='store_true')
parser.add_argument(
'--debug', help='Print debugging info', action='store_true')
parser.add_argument(
'--in_place_transitions',
help='Use in-place Transition array to determine Active Transitions',
action="store_true")
parser.add_argument(
'--optimize_candidates',
help='Optimize the candidate transitions',
action='store_true')
parser.add_argument('--zone', help='Name of time zone', required=True)
parser.add_argument('--year', help='Year of interest', type=int)
parser.add_argument('--date', help='DateTime of interest')
args = parser.parse_args()
# Configure logging
logging.basicConfig(level=logging.INFO)
# Find the zone.
zone_info = cast(ZoneInfo, zone_infos.ZONE_INFO_MAP.get(args.zone))
if not zone_info:
logging.error("Zone '%s' not found", args.zone)
sys.exit(1)
# Create the ZoneSpecifier for zone
zone_specifier = ZoneSpecifier(
zone_info_data=zone_info,
viewing_months=args.viewing_months,
debug=args.debug,
in_place_transitions=args.in_place_transitions,
optimize_candidates=args.optimize_candidates)
if args.year:
zone_specifier.init_for_year(args.year)
if args.debug:
logging.info('==== Final matches and transitions')
zone_specifier.print_matches_and_transitions()
elif args.date:
dt: datetime = datetime.strptime(args.date, "%Y-%m-%dT%H:%M")
if args.transition:
transition = zone_specifier.get_transition_for_datetime(dt)
if transition:
logging.info(transition)
else:
logging.error('Transition not found')
else:
offset_info = zone_specifier.get_timezone_info_for_datetime(dt)
if not offset_info:
logging.info('Invalid time')
else:
logging.info(
'%s (%s)',
to_utc_string(
offset_info.utc_offset,
offset_info.dst_offset,
),
offset_info.abbrev,
)
else:
print("One of --year or --date must be provided")
sys.exit(1)
