Saving Data with a Database¶

import os
import sqlite3


class DataBaseController:
    def __init__(self, dbase_file_name: str):
        app_file_path = os.path.join(os.path.dirname(__file__), dbase_file_name)
        self.app_db = sqlite3.connect(app_file_path)
        self.app_cursor = self.app_db.cursor()

    def close(self):
        self.app_db.close()

    def create_scores_table(self):
        """
        Creates the Scores table, if it doesn't exist yet
        """
        self.app_cursor.execute(
            """
                CREATE TABLE IF NOT EXISTS Scores (
                    name TEXT,
                    score INTEGER
                )
            """
        )
        self.app_db.commit()

    def add_score(self, name, score):
        """
        Saves a player's initials and score
        """
        self.app_cursor.execute(
            """
                INSERT INTO Scores (name, score)
                VALUES (:name, :score)
            """,
            {'name': name, 'score': score}
        )
        self.app_db.commit()

    def get_top_scores(self, count):
        """
        Returns the highest scores, best first
        """
        self.app_cursor.execute(
            """
                SELECT name, score
                FROM Scores
                ORDER BY score DESC
                LIMIT :count
            """,
            {'count': count}
        )
        return self.app_cursor.fetchall()

    # - Set the order of the rooms - #
    levels = ["WelcomeScreen", "DifficultySelect", "ShipSelect", "GamePlay", "HighScores"]

from GameFrame import EntryTextObject, Globals, DataBaseController
import pygame

class HighScoreEntry(EntryTextObject):
    """
    Lets the player type their initials, then saves their score
    """
    def __init__(self, room, x, y):
        """
        Initialise the initials entry box
        """
        EntryTextObject.__init__(self, room, x, y, max_len=3)

        # set values
        self.size = 60
        self.font = 'Arial Black'
        self.colour = (255,255,0)
        self.update_text()
        self.saved = False

    def key_pressed(self, key):
        """
        Types the initials, and saves them when Enter is pressed
        """
        if self.saved:
            return

        EntryTextObject.key_pressed(self, key)
        if key[pygame.K_RETURN] and len(self.text) > 0:
            self.save_score()

    def save_score(self):
        """
        Saves the score in the database, then shows the top scores
        """
        self.saved = True
        database = DataBaseController("scores.db")
        database.create_scores_table()
        database.add_score(self.text, Globals.SCORE)
        top_scores = database.get_top_scores(5)
        database.close()
        self.room.show_scores(top_scores)

from Objects.Title import Title
from Objects.Ship import Ship
from Objects.Zork import Zork
from Objects.Asteroid import Asteroid
from Objects.Laser import Laser
from Objects.Astronaut import Astronaut
from Objects.Hud import Score, Lives, Rescued, Streak, PowerMeter
from Objects.DifficultyMenu import DifficultyMenu
from Objects.Bonus import RepairKit, Shield
from Objects.ShipMenu import ShipMenu
from Objects.HighScoreEntry import HighScoreEntry

from GameFrame import Level, Globals, TextObject
from Objects.HighScoreEntry import HighScoreEntry

class HighScores(Level):
    """
    Screen for saving the player's score and showing the top scores
    """
    def __init__(self, screen, joysticks):
        Level.__init__(self, screen, joysticks)

        # set background image
        self.set_background_image("background.png")

        # show the player's score and ask for their initials
        self.message = TextObject(self, 380, 120, f"Your score: {Globals.SCORE}", 50, 'Arial Black', (255,255,255))
        self.add_room_object(self.message)
        self.prompt = TextObject(self, 380, 220, "Type your initials, then press Enter", 30, 'Arial Black', (255,255,255))
        self.add_room_object(self.prompt)

        # add the initials entry box
        self.add_room_object(HighScoreEntry(self, 580, 290))

    def show_scores(self, top_scores):
        """
        Shows the top scores, then goes back to the welcome screen
        """
        self.prompt.text = "Top scores"
        self.prompt.update_text()

        # one line of text for each score
        y = 400
        for name, score in top_scores:
            line = TextObject(self, 540, y, f"{name}   {score}", 40, 'Arial Black', (255,255,255))
            self.add_room_object(line)
            y += 50

        # go back to the welcome screen after 5 seconds
        self.set_timer(150, self.finish)

    def finish(self):
        """
        Ends this Room so the game goes back to the welcome screen
        """
        self.running = False

from Rooms.WelcomeScreen import WelcomeScreen
from Rooms.GamePlay import GamePlay
from Rooms.DifficultySelect import DifficultySelect
from Rooms.ShipSelect import ShipSelect
from Rooms.HighScores import HighScores