def test_get_transition_for_datetime(self) -> None: zone_specifier = ZoneSpecifier( cast(ZoneInfo, zone_infos.ZONE_INFO_America_Los_Angeles), viewing_months=14, ) # 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", default=True) parser.add_argument('--no_in_place_transitions', help='Disable --in_place_transitions', action="store_false", dest='in_place_transitions') parser.add_argument('--optimize_candidates', help='Optimize the candidate transitions', action='store_true', default=True) parser.add_argument('--no_optimize_candidates', help='Disable --optimize_candidates', action='store_false', dest='optimize_candidates') 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)