Final exam version

solar_objects.py

@@ -0,0 +1,89 @@
+import math
+import random
+
+
+class SpaceObject:
+    def __init__(self, x, y, radius, color):
+        self.x = x
+        self.y = y
+        self.radius = radius
+        self.color = color
+
+    def update(self):
+        pass
+
+    def draw(self, screen, pygame):
+        pass
+
+
+class Star(SpaceObject):
+    def __init__(self, x, y, radius=34, color=(255, 185, 40), name="Star"):
+        super().__init__(x, y, radius, color)
+        self.name = name
+        self.planets = []
+
+    def add_planet(self, planet):
+        self.planets.append(planet)
+
+    def set_position(self, x, y):
+        self.x = x
+        self.y = y
+
+    def update(self):
+        for planet in self.planets:
+            planet.update()
+
+    def draw(self, screen, pygame, show_orbits=True):
+        if show_orbits:
+            for planet in self.planets:
+                planet.draw_orbit(screen, pygame)
+
+        pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), self.radius)
+        pygame.draw.circle(screen, (255, 230, 120), (int(self.x), int(self.y)), self.radius // 2)
+
+        for planet in self.planets:
+            planet.draw(screen, pygame)
+
+
+class Planet(SpaceObject):
+    def __init__(self, star, orbit_number, orbit_radius, angle, speed, direction, radius=None, color=None):
+        self.star = star
+        self.orbit_number = orbit_number
+        self.orbit_radius = orbit_radius
+        self.angle = angle
+        self.speed = speed
+        self.direction = direction
+
+        if radius is None:
+            radius = random.randint(8, 13)
+
+        if color is None:
+            color = (
+                random.randint(40, 220),
+                random.randint(70, 230),
+                random.randint(90, 255),
+            )
+
+        x, y = self.calculate_position()
+        super().__init__(x, y, radius, color)
+
+    def calculate_position(self):
+        x = self.star.x + math.cos(self.angle) * self.orbit_radius
+        y = self.star.y + math.sin(self.angle) * self.orbit_radius
+        return x, y
+
+    def update(self):
+        self.angle += self.speed * self.direction
+        self.x, self.y = self.calculate_position()
+
+    def draw_orbit(self, screen, pygame):
+        pygame.draw.circle(
+            screen,
+            (160, 160, 160),
+            (int(self.star.x), int(self.star.y)),
+            int(self.orbit_radius),
+            1,
+        )
+
+    def draw(self, screen, pygame):
+        pygame.draw.circle(screen, self.color, (int(self.x), int(self.y)), self.radius)