voting_system.py

class VotingSystem:
def __init__(self):
self.candidates = {}
self.votes = {}
self.voters = set()
def register_candidate(self, candidate_id, candidate_name):
if candidate_id in self.candidates:
print("Candidate already registered!")
else:
self.candidates[candidate_id] = candidate_name
self.votes[candidate_id] = 0
print(f"Candidate '{candidate_name}' registered successfully.")
def cast_vote(self, voter_id, candidate_id):
if voter_id in self.voters:
print("You have already voted.")
return
if candidate_id not in self.candidates:
print("Invalid candidate ID.")
return
self.voters.add(voter_id)
self.votes[candidate_id] += 1
print(f"Vote casted for {self.candidates[candidate_id]} by voter {voter_id}.")
def display_results(self):
print("\n--- Voting Results ---")
for candidate_id, vote_count in self.votes.items():
print(f"{self.candidates[candidate_id]}: {vote_count} votes")
def is_valid_candidate(self, candidate_name):
return len(candidate_name.strip()) > 0
def main():
system = VotingSystem()
while True:
print("\n--- Voting System ---")
print("1. Register Candidate")
print("2. Cast Vote")
print("3. Display Results")
print("4. Exit")
choice = input("Enter your choice: ")
if choice == '1':
candidate_id = input("Enter candidate ID: ")
candidate_name = input("Enter candidate name: ")
if system.is_valid_candidate(candidate_name):
system.register_candidate(candidate_id, candidate_name)
else:
print("Invalid candidate name. Name must not be empty.")
elif choice == '2’:
voter_id = input("Enter your voter ID: ")
print("Available candidates:")
for candidate_id, candidate_name in
system.candidates.items():
print(f"{candidate_id}:{candidate_name}") candidate_id = input("Enter candidate ID to vote for: ")
system.cast_vote(voter_id, candidate_id)
elif choice == '3’:
system.display_results()
elif choice == '4’:
print("Exiting voting system.")
break
else:
print("Invalid choice. Please try again.")
if __name__ == "__main__":
main()