Kgshop commited on
Commit
dec5f73
·
verified ·
1 Parent(s): 072f33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +102 -68
app.py CHANGED
@@ -1,7 +1,6 @@
1
- from flask import Flask, render_template, request, jsonify
2
  import json
3
  from pywebpush import webpush
4
- # from py_vapid import VAPID <-- УДАЛИТЕ ЭТУ СТРОКУ
5
 
6
  app = Flask(__name__)
7
  PORT = 7860
@@ -72,68 +71,8 @@ def send_push_to_all(message_body):
72
  save_subscriptions(valid_subscriptions) # Сохраняем только валидные подписки
73
 
74
 
75
- # --- Flask маршруты ---
76
-
77
- @app.route('/')
78
- def index():
79
- """Главная страница, отображает новости и запрашивает разрешение на уведомления."""
80
- news = load_news()
81
- public_key = VAPID_PUBLIC_KEY_BASE64 # Передаем публичный ключ в шаблон
82
- return render_template('index.html', news=news, public_key=public_key)
83
-
84
- @app.route('/add_news', methods=['POST'])
85
- def add_news():
86
- """API endpoint для добавления новой новости."""
87
- news_text = request.form.get('news_text')
88
- if news_text:
89
- news_list = load_news()
90
- news_list.append({"text": news_text})
91
- save_news(news_list)
92
-
93
- # Отправляем push уведомления о новой новости
94
- send_push_to_all(news_text)
95
-
96
- return jsonify({"message": "Новость добавлена и уведомления отправлены"}), 200
97
- return jsonify({"error": "Текст новости не предоставлен"}), 400
98
-
99
- @app.route('/get_news')
100
- def get_news():
101
- """API endpoint для получения списка новостей."""
102
- news = load_news()
103
- return jsonify(news)
104
-
105
- @app.route('/save_subscription', methods=['POST'])
106
- def save_subscription():
107
- """API endpoint для сохранения подписки пользователя на уведомления."""
108
- subscription_data = request.get_json()
109
- if subscription_data:
110
- subscriptions = load_subscriptions()
111
- subscriptions.append(subscription_data)
112
- save_subscriptions(subscriptions)
113
- print(f"Подписка сохранена: {subscription_data['endpoint']}") # Логирование для отладки
114
- return jsonify({"message": "Подписка сохранена"}), 201
115
- return jsonify({"error": "Неверные данные подписки"}), 400
116
-
117
- @app.route('/service-worker.js')
118
- def service_worker():
119
- """Маршрут для service-worker.js."""
120
- return app.send_static_file('service-worker.js')
121
-
122
- @app.route('/manifest.json')
123
- def manifest():
124
- """Маршрут для manifest.json."""
125
- return app.send_static_file('manifest.json')
126
-
127
-
128
- if __name__ == '__main__':
129
- app.run(debug=True, port=PORT, host='0.0.0.0') # Доступно извне и на порту 7860
130
-
131
-
132
- # --- HTML шаблоны и статические файлы ---
133
-
134
- # Создайте папку 'templates' и файл 'index.html' внутри нее
135
- # templates/index.html:
136
- """
137
  <!DOCTYPE html>
138
  <html lang="ru">
139
  <head>
@@ -151,9 +90,6 @@ if __name__ == '__main__':
151
  <h1>Новости</h1>
152
 
153
  <ul class="list-group" id="news-list">
154
- {% for item in news %}
155
- <li class="list-group-item">{{ item.text }}</li>
156
- {% endfor %}
157
  </ul>
158
 
159
  <hr>
@@ -169,7 +105,7 @@ if __name__ == '__main__':
169
  </div>
170
 
171
  <script>
172
- const publicKey = '{{ public_key }}'; // Передаем публичный ключ из Flask
173
 
174
  document.addEventListener('DOMContentLoaded', () => {
175
  requestNotificationPermission();
@@ -317,3 +253,101 @@ if __name__ == '__main__':
317
  </html>
318
  """
319
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify, Response
2
  import json
3
  from pywebpush import webpush
 
4
 
5
  app = Flask(__name__)
6
  PORT = 7860
 
71
  save_subscriptions(valid_subscriptions) # Сохраняем только валидные подписки
72
 
73
 
74
+ # --- HTML для главной страницы ---
75
+ HTML_INDEX = """
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
76
  <!DOCTYPE html>
77
  <html lang="ru">
78
  <head>
 
90
  <h1>Новости</h1>
91
 
92
  <ul class="list-group" id="news-list">
 
 
 
93
  </ul>
94
 
95
  <hr>
 
105
  </div>
106
 
107
  <script>
108
+ const publicKey = '%PUBLIC_KEY%'; // Заменится на реальный публичный ключ
109
 
110
  document.addEventListener('DOMContentLoaded', () => {
111
  requestNotificationPermission();
 
253
  </html>
254
  """
255
 
256
+ # --- Service Worker код ---
257
+ SERVICE_WORKER_JS = """
258
+ self.addEventListener('push', event => {
259
+ const message = event.data.text();
260
+ const title = 'Новая новость!';
261
+ const options = {
262
+ body: message,
263
+ icon: '/no-icon.png' // Путь к иконке (теперь в корне, или можно убрать если не нужна иконка)
264
+ };
265
+
266
+ event.waitUntil(self.registration.showNotification(title, options));
267
+
268
+ // Отправляем сообщение клиенту (странице) о получении push уведомления
269
+ self.clients.matchAll({ type: 'window' }).then(clients => {
270
+ clients.forEach(client => {
271
+ client.postMessage({ type: 'push-received', message: message });
272
+ });
273
+ });
274
+ });
275
+
276
+ self.addEventListener('notificationclick', event => {
277
+ event.notification.close();
278
+ event.waitUntil(clients.openWindow('/')); // Открываем главное окно при клике на уведомление
279
+ });
280
+ """
281
+
282
+ # --- Manifest JSON ---
283
+ MANIFEST_JSON = """
284
+ {
285
+ "name": "PWA Новости",
286
+ "short_name": "Но��ости",
287
+ "start_url": "/",
288
+ "display": "standalone",
289
+ "background_color": "#fff",
290
+ "theme_color": "#007bff"
291
+ }
292
+ """
293
+
294
+
295
+ # --- Flask маршруты ---
296
+
297
+ @app.route('/')
298
+ def index():
299
+ """Главная страница, отображает новости и запрашивает разрешение на уведомления."""
300
+ news = load_news()
301
+ public_key = VAPID_PUBLIC_KEY_BASE64 # Передаем публичный ключ в шаблон
302
+ html = HTML_INDEX.replace('%PUBLIC_KEY%', public_key)
303
+ # Заполняем список новостей динамически на сервере, чтобы избежать flicker при загрузке
304
+ news_list_html = ''.join([f'<li class="list-group-item">{item["text"]}</li>' for item in news])
305
+ html = html.replace('<ul class="list-group" id="news-list"></ul>', f'<ul class="list-group" id="news-list">{news_list_html}</ul>')
306
+ return html
307
+
308
+ @app.route('/add_news', methods=['POST'])
309
+ def add_news():
310
+ """API endpoint для добавления новой новости."""
311
+ news_text = request.form.get('news_text')
312
+ if news_text:
313
+ news_list = load_news()
314
+ news_list.append({"text": news_text})
315
+ save_news(news_list)
316
+
317
+ # Отправляем push уведомления о новой новости
318
+ send_push_to_all(news_text)
319
+
320
+ return jsonify({"message": "Новость добавлена и уведомления отправлены"}), 200
321
+ return jsonify({"error": "Текст новости не предоставлен"}), 400
322
+
323
+ @app.route('/get_news')
324
+ def get_news():
325
+ """API endpoint для получения списка новостей."""
326
+ news = load_news()
327
+ return jsonify(news)
328
+
329
+ @app.route('/save_subscription', methods=['POST'])
330
+ def save_subscription():
331
+ """API endpoint для сохранения подписки пользователя на уведомления."""
332
+ subscription_data = request.get_json()
333
+ if subscription_data:
334
+ subscriptions = load_subscriptions()
335
+ subscriptions.append(subscription_data)
336
+ save_subscriptions(subscriptions)
337
+ print(f"Подписка сохранена: {subscription_data['endpoint']}") # Логирование для отладки
338
+ return jsonify({"message": "Подписка сохранена"}), 201
339
+ return jsonify({"error": "Неверные данные подписки"}), 400
340
+
341
+ @app.route('/service-worker.js')
342
+ def service_worker():
343
+ """Маршрут для service-worker.js."""
344
+ return Response(SERVICE_WORKER_JS, mimetype='application/javascript')
345
+
346
+ @app.route('/manifest.json')
347
+ def manifest():
348
+ """Маршрут для manifest.json."""
349
+ return Response(MANIFEST_JSON, mimetype='application/json')
350
+
351
+
352
+ if __name__ == '__main__':
353
+ app.run(debug=True, port=PORT, host='0.0.0.0') # Доступно извне и на порту 7860