prettify and lint optimization

This commit is contained in:
2023-12-10 17:24:14 +01:00
parent 274c69d0c4
commit 4cd62ef794
32 changed files with 599 additions and 262 deletions

View File

@@ -1,21 +1,24 @@
def parse_card_data(line):
"""Parse a single line of card data into card number, winning numbers, and own numbers."""
try:
card_info, number_parts = line.split(':')
winning_part, own_part = number_parts.split('|')
card_info, number_parts = line.split(":")
winning_part, own_part = number_parts.split("|")
# Extract card number from card_info
card_number = ''.join(filter(str.isdigit, card_info))
card_number = "".join(filter(str.isdigit, card_info))
# Convert number strings to integer lists
winning_numbers = set(map(int, winning_part.split()))
own_numbers = list(map(int, own_part.split()))
print(f"Card {card_number}: Winning Numbers: {winning_numbers}, Own Numbers: {own_numbers}")
print(
f"Card {card_number}: Winning Numbers: {winning_numbers}, Own Numbers: {own_numbers}"
)
return winning_numbers, own_numbers
except ValueError as e:
raise ValueError(f"Error parsing line '{line}': {e}")
def calculate_card_points(winning_numbers, own_numbers):
"""Calculate the points for a single card."""
points = 0
@@ -24,22 +27,27 @@ def calculate_card_points(winning_numbers, own_numbers):
points = 1 if points == 0 else points * 2
return points
def process_cards(file_path):
"""Process all cards in the given file and return the total points."""
total_points = 0
with open(file_path, 'r') as file:
with open(file_path, "r") as file:
for line in file:
winning_numbers, own_numbers = parse_card_data(line.strip())
total_points += calculate_card_points(winning_numbers, own_numbers)
return total_points
def test():
"""Run tests using the test.txt file."""
expected_result = 13 # Based on the given example
result = process_cards('../test.txt')
assert result == expected_result, f"Test failed: Expected {expected_result}, got {result}"
result = process_cards("../test.txt")
assert (
result == expected_result
), f"Test failed: Expected {expected_result}, got {result}"
print(f"Test passed: {result} points")
def main():
"""Main function to process the input file and display results."""
try:
@@ -47,11 +55,12 @@ def main():
test()
# Process actual input
total_points = process_cards('../input.txt')
total_points = process_cards("../input.txt")
print(f"Total points from input.txt: {total_points}")
except Exception as e:
print(f"An error occurred: {e}")
if __name__ == "__main__":
main()

View File

@@ -1,25 +1,26 @@
def parse_card_data(line):
"""Parse a single line of card data into card number, winning numbers, and own numbers."""
try:
card_info, number_parts = line.split(':')
winning_part, own_part = number_parts.split('|')
card_number = ''.join(filter(str.isdigit, card_info))
card_info, number_parts = line.split(":")
winning_part, own_part = number_parts.split("|")
card_number = "".join(filter(str.isdigit, card_info))
winning_numbers = set(map(int, winning_part.split()))
own_numbers = list(map(int, own_part.split()))
return int(card_number), winning_numbers, own_numbers
except ValueError as e:
raise ValueError(f"Error parsing line '{line}': {e}")
def calculate_matches(winning_numbers, own_numbers):
"""Calculate the number of matches for a single card."""
return sum(1 for number in own_numbers if number in winning_numbers)
def process_cards(file_path):
"""Process all cards in the given file and return the total number of cards."""
with open(file_path, 'r') as file:
with open(file_path, "r") as file:
cards = [parse_card_data(line.strip()) for line in file]
total_cards = len(cards) # Start with original cards
i = 0
while i < len(cards):
card_number, winning_numbers, own_numbers = cards[i]
@@ -32,13 +33,17 @@ def process_cards(file_path):
return len(cards)
def test():
"""Run tests using the test.txt file."""
expected_result = 30 # Based on the given example
result = process_cards('../test.txt')
assert result == expected_result, f"Test failed: Expected {expected_result}, got {result}"
result = process_cards("../test.txt")
assert (
result == expected_result
), f"Test failed: Expected {expected_result}, got {result}"
print(f"Test passed: {result} cards")
def main():
"""Main function to process the input file and display results."""
try:
@@ -46,11 +51,12 @@ def main():
test()
# Process actual input
total_cards = process_cards('../input.txt')
total_cards = process_cards("../input.txt")
print(f"Total cards from input.txt: {total_cards}")
except Exception as e:
print(f"An error occurred: {e}")
if __name__ == "__main__":
main()