最后用if__name__ == '__main__': main()
时间: 2023-07-21 22:38:08 浏览: 43
好的,下面是加入 `if __name__ == '__main__':` 的完整代码:
```python
import sqlite3
def create_table():
conn = sqlite3.connect('courses.db')
c = conn.cursor()
# 创建数据表
c.execute('''CREATE TABLE courses
(name TEXT, teacher TEXT, location TEXT, weekday TEXT, start_time INT, end_time INT)''')
conn.commit()
conn.close()
def add_course(name, teacher, location, weekday, start_time, end_time):
conn = sqlite3.connect('courses.db')
c = conn.cursor()
# 添加一门课程
c.execute(f"INSERT INTO courses VALUES ('{name}', '{teacher}', '{location}', '{weekday}', {start_time}, {end_time})")
conn.commit()
conn.close()
def view_courses():
conn = sqlite3.connect('courses.db')
c = conn.cursor()
# 查询所有课程信息
c.execute("SELECT * FROM courses")
rows = c.fetchall()
for row in rows:
print(row)
conn.close()
def edit_course(name, teacher, location, weekday, start_time, end_time):
conn = sqlite3.connect('courses.db')
c = conn.cursor()
# 修改一门课程
c.execute(f"UPDATE courses SET teacher='{teacher}', location='{location}', weekday='{weekday}', start_time={start_time}, end_time={end_time} WHERE name='{name}'")
conn.commit()
conn.close()
def delete_course(name):
conn = sqlite3.connect('courses.db')
c = conn.cursor()
# 删除一门课程
c.execute(f"DELETE FROM courses WHERE name='{name}'")
conn.commit()
conn.close()
def main():
# 创建数据表
create_table()
# 添加一门课程
add_course('计算机网络', '张三', '教学楼101', '星期一', 8, 10)
# 查看所有课程
view_courses()
# 修改一门课程
edit_course('计算机网络', '李四', '教学楼201', '星期二', 10, 12)
# 查看所有课程
view_courses()
# 删除一门课程
delete_course('计算机网络')
# 查看所有课程
view_courses()
if __name__ == '__main__':
main()
```
这个代码实现了创建数据表、添加课程、查看课程、编辑课程、删除课程等功能,并使用了 `if __name__ == '__main__':` 来保证了代码的可重用性和可扩展性。你可以根据需要进行修改和扩展,让其更加符合你的实际需求。