From cd42352303043d30bc462961036e7d9056860ee1 Mon Sep 17 00:00:00 2001
From: DIVYASREE S <divyasrees.cse.41@gmail.com>
Date: Thu, 10 Oct 2024 22:14:01 +0530
Subject: [PATCH] Added proof_of_work.py

---
 blockchain/proof_of_work.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 blockchain/proof_of_work.py

diff --git a/blockchain/proof_of_work.py b/blockchain/proof_of_work.py
new file mode 100644
index 000000000..4bf14106b
--- /dev/null
+++ b/blockchain/proof_of_work.py
@@ -0,0 +1,31 @@
+import hashlib
+import time
+
+def proof_of_work(difficulty: int) -> int:
+  """
+  Simulates a Proof of Work mining process.
+  
+  The miner must find a nonce such that the hash of the nonce starts 
+  with a specific number of leading zeros (difficulty)
+  
+  Args:
+        difficulty (int): The number of leading zeros required in the hash.
+  Returns:
+        int: The nonce value that solves the puzzle.
+        
+  Example:
+        >>> result = proof_of_work(2)  # Difficulty of 2 should be fast
+        >>> isinstance(result, int)
+        True
+    """
+  prefix = '0' * difficulty
+  nonce = 0
+  start = time.time()  # Timing starts
+
+  while True:
+      hash_result = hashlib.sha256(f"{nonce}".encode()).hexdigest()
+      if hash_result.startswith(prefix):
+          end = time.time()  # Timing ends
+          print(f"Time taken: {end - start:.2f}s")  # Print time taken
+          return nonce
+      nonce += 1