From 3d94989c9b2392f16132028158be2817b1bc56e8 Mon Sep 17 00:00:00 2001
From: Nikolya Andreychik <kolya_andreychik@mail.ru>
Date: Tue, 6 Apr 2021 03:44:19 -0700
Subject: [PATCH] Crypted leaderboard added, first record added too

---
 EndSceen.py |  64 +++++++++++++++++++++++++++++++++-------------------
 leaders.lb  | Bin 0 -> 138 bytes
 leaders.txt |   7 ------
 3 files changed, 41 insertions(+), 30 deletions(-)
 create mode 100644 leaders.lb
 delete mode 100644 leaders.txt

diff --git a/EndSceen.py b/EndSceen.py
index 360e43b..9bacafe 100644
--- a/EndSceen.py
+++ b/EndSceen.py
@@ -2,8 +2,10 @@
 Enemy entity class
 """
 
-
+import pickle
 import pygame
+import hashlib
+from cryptography.fernet import Fernet
 
 
 class EndSceen():
@@ -22,7 +24,11 @@ class EndSceen():
             mainGameClass.getFont(), 15
         )
 
+        key = b'Lh2b2rragfwD8QR4VU-V2TmSuio4yp-WbFwo4tcoyzs='
+        self.code = Fernet(key)
+
         self.game = mainGameClass
+        self.saveFileName = 'leaders.lb'
 
     def newEndScreen(self):
         self.endScreenTimer = 0;
@@ -30,23 +36,29 @@ class EndSceen():
 
         self.scoresFromFile = []
         self.data = []
+        self.sortedDataByScores = []
 
         self.isBackButton = True
 
-        fileWithData = open('leaders.txt')
+        try:
+            fileWithData = open(self.saveFileName, 'rb')
+        except IOError as e:
+            pass
+        else:
+            listPlayers = pickle.load(fileWithData)
 
-        for line in fileWithData:
-            oneStr = line.split()
+            for line in listPlayers:
+                oneStr = self.code.decrypt(line).decode().split()
 
-            if (len(oneStr) == 3):
-                self.data.append(
-                    [oneStr[0][:10], int(oneStr[1]), float(oneStr[2])]
-                )
+                if (len(oneStr) == 3):
+                    self.data.append(
+                        [oneStr[0][:10], int(oneStr[1]), float(oneStr[2])]
+                    )
 
-        fileWithData.close()
+            fileWithData.close()
 
-        self.sortedDataByScores = sorted(enumerate(self.data),
-                                        key=lambda i: i[1][1], reverse=True)
+            self.sortedDataByScores = sorted(enumerate(self.data),
+                                            key=lambda i: i[1][1], reverse=True)
 
 
     def renderText(self, text, font, color, center, backColor=None):
@@ -199,19 +211,25 @@ class EndSceen():
                     self.playerName += pygame.key.name(event.key).lower()
 
     def saveResults(self):
-        with open ('leaders.txt', 'r') as fileWithData:
-            tmpData = fileWithData.readlines()
-        fileWithData.close()
-
         newData = []
-        for line in tmpData:
-            if len(line.split()) == 3:
-                if not (line.split()[0].rstrip() == self.playerName.rstrip()):
-                    newData.append(line)
 
-        newData.append('{0} {1} {2:.2f}\n'.format(self.playerName,
-                        self.game.getScore(), self.game.getTime()))
+        try:
+            fileWithData = open(self.saveFileName, 'rb')
+        except IOError as e:
+            pass
+        else:
+            tmpData = pickle.load(fileWithData)
+            fileWithData.close()
 
-        with open ('leaders.txt', 'w') as fileWithData:
-            fileWithData.writelines(newData)
+            for line in tmpData:
+                oneStr = self.code.decrypt(line).decode()
+                if len(oneStr.split()) == 3:
+                    if not (oneStr.split()[0].rstrip() == self.playerName.rstrip()):
+                        newData.append(line)
+
+        newData.append(self.code.encrypt(('{0} {1} {2:.2f}\n'.format(self.playerName,
+                        self.game.getScore(), self.game.getTime())).encode()))
+
+        with open (self.saveFileName, 'wb') as fileWithData:
+            pickle.dump(newData, fileWithData)
         fileWithData.close()
diff --git a/leaders.lb b/leaders.lb
new file mode 100644
index 0000000000000000000000000000000000000000..1dd5a2ced796bbd4bb0a4c6f5569302775264925
GIT binary patch
literal 138
zcmZo*nOe^P0kKn@E7Bc7z$rb+rPwPkBrvVetTfd-yf~o3*V8r4HQ&IwwAk6WTsJht
zAUQSO*`O%U(=#$K-cL6vG|b;9J2EdP-Pg#~)FUO-!mTL1%+JTAGB?;g!ZRbsxx&)j
iG{qx3%0JgAsx+{wDk;pnu%a+0(=*7(tJro*q8<Q<?JTwc

literal 0
HcmV?d00001

diff --git a/leaders.txt b/leaders.txt
deleted file mode 100644
index 018afc3..0000000
--- a/leaders.txt
+++ /dev/null
@@ -1,7 +0,0 @@
-Kolya 1 11.37
-Coolguy 1 10.75
-Wow 74 185.72
-Cheater 1000 0.01
-New-Kolya 2 15.20
-1 1 10.02
-RUSlan 29 82.42