Fully Implemented Saving Results

README.md

@@ -15,7 +15,7 @@ The launcher `pylaunch.py` is best used on Linux or with WSL, unless your Python
 - No contests and disqualification finishes
 - Specify whether a wrestler wins by pinfall, submission or knockout
 - Supports booking Tag Team and Singles matches as of right now
-- Allows for saving match results to a text file (under construction, attendance/views do not show as of right now)
+- Allows for saving match results to a text file 
 
 # Screenshot of Latest Version
 ![image](https://i.imgur.com/j9SIR2Y.png)

pywrbs.py

@@ -1,6 +1,6 @@
 import random
 
-def writeresults(fed, eventna, finalgrade, results):
+def writeresults(fed, eventna, finalgrade, results, viewreport):
     filename = input("Enter a name for a text file to save these results into (omit the .txt extension)\n")
     resultfile = open(filename + ".txt","a")
     resultfile.write("\n************************************\n"+fed+" " +eventna+" RESULTS\n************************************\n") 
@@ -11,6 +11,7 @@ def writeresults(fed, eventna, finalgrade, results):
         i+=1
 
     resultfile.write("\nThe final rating for " +fed +" " +eventna + " is: " +finalgrade + "\n")
+    resultfile.write(viewreport)
     print("The file '" + filename + ".txt'" + " has been saved to successfully.")
 
 def getwinmethod(wm):
@@ -33,27 +34,29 @@ def tvratings(eventna,fed):
     if rawstring in eventna.lower() and fed.lower() == "wwe":
         viewers = random.randint(1500000,3000000)
         # using f strings to format the viewer number with commas
-        print("\nUSA Network TV Viewers: " +f'{viewers:,}')
+        viewreport = "\nUSA Network TV Viewers: " +f'{viewers:,}'
 
     elif smackstring in eventna.lower() and fed.lower() == "wwe":
         viewers = random.randint(1000000,2800000)
-        print("\nFOX TV Viewers: "+f'{viewers:,}')
+        viewreport = "\nFOX TV Viewers: "+f'{viewers:,}'
 
     elif rampagestring in eventna.lower() and fed.lower() == "aew" or fed.lower() == "wcw" and nitrostring in eventna.lower():
         viewers = random.randint(430000,1100000)  
-        print("\nTNT TV Viewers: "+f'{viewers:,}')
+        viewreport = "\nTNT TV Viewers: "+f'{viewers:,}'
 
     elif dynamitestring in eventna.lower() and fed.lower() == "aew":
         viewers = random.randint(500000,1400000)
-        print("\nTBS TV Viewers: "+f'{viewers:,}')
+        viewreport = "\nTBS TV Viewers: "+f'{viewers:,}'
 
     elif darkstring in eventna.lower() and fed.lower() == "aew":
         viewers = random.randint(300000, 1800000)
-        print("\nYouTube Viewers: "+f'{viewers:,}')
+        viewreport = "\nYouTube Viewers: "+f'{viewers:,}'
 
     else:
         viewers = random.randint(1000,20000)
-        print("\nLive Event Attendance: " +f'{viewers:,}')
+        viewreport = "\nLive Event Attendance: " +f'{viewers:,}'
+
+    return viewreport
 
 def overnesseval(over, wrestler,matchty):
     if over == 1:
@@ -250,11 +253,12 @@ def matchtime():
     finalrating = int(finalrating / matches)
     finalgrade = ratingtograde(finalrating)
     print("The final rating for " +fed +" " +eventna + " is: " +finalgrade)
-    tvratings(eventna,fed)
+    viewreport = tvratings(eventna,fed)
+    print(viewreport)
     saveresults = input("Would you like to save the show results?(y/n)\n")
 
     if saveresults.lower() == "y":
-        writeresults(fed, eventna, finalgrade, results)
+        writeresults(fed, eventna, finalgrade, results, viewreport)
 
 except ValueError:
     print("An error occurred, invalid input...\nPyWRBS shutting down...")