写一段生成烟花的完整python代码

import pygame
import math

# 初始化pygame
pygame.init()

# 创建屏幕
screen = pygame.display.set_mode((800, 600))

# 加载背景图片
background = pygame.image.load('background.jpg')

# 加载烟花图片
firework = pygame.image.load('firework.png')

# 加载音乐
pygame.mixer.music.load('bg_music.mp3')

# 播放音乐
pygame.mixer.music.play(-1)

# 创建烟花类
class Firework:

# 初始化方法
def __init__(self):
# 初始位置
self.x = 400
self.y = 580
# 角度
self.angle = 45
# 速度
self.v_x = 10 * math.cos(self.angle * math.pi / 180)
self.v_y = 10 * math.sin(self.angle * math.pi / 180)

# 更新烟花位置
def update(self):
self.x += self.v_x
self.y -= self.v_y
self.v_y -= 0.3

pygame.time.delay(1000)

# 创建烟花实例
firework = Firework()

# 游戏循环
while True:
for event in pygame.event.get():
# 点击右上角的退出按钮
if event.type == pygame.QUIT:
# 退出
pygame.quit()
exit()

# 加载背景图片
screen.blit(background, (0, 0))

# 更新烟花位置
firework.update()

# 加载烟花
screen.blit(firework, (firework.x, firework.y))

# 刷新
pygame.display.update()