Create voting_system.py

Author: Balaji-P-A

voting_system.py

@@ -0,0 +1,60 @@
+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()