V1.30 notification

aitrainer_backoffice/controlling/admin/customer.py

@@ -10,10 +10,7 @@ from django.urls import path
 
 from ..models.customer import Customer
 from ..models.customer import Sport
-from ..mautic import MauticHelper
-from ..cron import cron
-from ..push_notification import messaging
-
+from ..automation.notification import Notification
 
 class SportFilter(SimpleListFilter, ABC):
     title = "Sport"
@@ -22,7 +19,7 @@ class SportFilter(SimpleListFilter, ABC):
     def lookups(self, request, model_admin):
         data = []
         for s in Sport.objects.filter():
-            data.append([s.sport_id, s.sport_name])
+            data.append([s.sport_id, s.name])
         return data
 
     def queryset(self, request, queryset):
@@ -33,6 +30,8 @@ class SportFilter(SimpleListFilter, ABC):
 
 
 class CustomerAdmin(admin.ModelAdmin):
+    notif = Notification()
+
     change_list_template = "controlling/mautic.html"
     list_display = ('customer_id', 'name','firstname', 'email', 'date_add', 'get_sport')
     list_filter = (
@@ -41,10 +40,10 @@ class CustomerAdmin(admin.ModelAdmin):
     )
 
     def get_sport(self, obj):
-        return obj.sport.sport_name
+        return obj.sport.name
 
     get_sport.short_description = 'Sport'
-    get_sport.admin_order_field = 'sport__sport_name'
+    get_sport.admin_order_field = 'sport__name'
 
 
     # If you would like to add a default range filter
@@ -75,7 +74,7 @@ class CustomerAdmin(admin.ModelAdmin):
         return my_urls + urls
 
     def set_mautic(self, request):
-        messaging.send_to_token()
+        self.notif.run()
         return HttpResponseRedirect("../")
 
 

aitrainer_backoffice/controlling/admin/frequent_exercises.py

@@ -1,6 +1,6 @@
 from django.contrib import admin
 
-from ..models import ExerciseType
+from aitrainer_backoffice.models.exercise_type import ExerciseType
 
 
 class FrequentExerciseTypeAdmin(admin.ModelAdmin):
@@ -11,5 +11,5 @@ class FrequentExerciseTypeAdmin(admin.ModelAdmin):
         return qs.filter(active=True)
 
 
-admin.site.register(ExerciseType, FrequentExerciseTypeAdmin)
-admin.autodiscover()
+#admin.site.register(FrequentExerciseTypeAdmin)
+#admin.autodiscover()

aitrainer_backoffice/controlling/automation/fcm.py

@@ -0,0 +1,97 @@
+import datetime
+from firebase_admin import messaging, initialize_app, exceptions
+
+class FCM:
+    logo_url = 'https://workouttest.com/wp-content/uploads/2020/10/WT_long_logo.png'
+
+    # default constructor
+    def __init__(self):
+        # To learn more, visit the docs here:
+        # https://cloud.google.com/docs/authentication/getting-started>
+        default_app = initialize_app()
+
+    def send_to_multiple_token(self, title, body, registration_token, image_url = None):
+        try: 
+            notification_image_url = image_url
+            #if image_url == None:
+            #    notification_image_url = self.logo_url
+
+            message = messaging.MulticastMessage(
+                notification=messaging.Notification(
+                    title=title,
+                    body=body,
+                ),
+                android=messaging.AndroidConfig(
+                    ttl=datetime.timedelta(seconds=3600),
+                    priority='normal',
+                    notification=messaging.AndroidNotification(
+                        image=notification_image_url,
+                    ),
+                ),
+                apns=messaging.APNSConfig(
+                    payload=messaging.APNSPayload(
+                        aps=messaging.Aps(badge=1),
+                    ),
+                    fcm_options= messaging.APNSFCMOptions(
+                        image=notification_image_url,
+                    )
+                ),
+                tokens= registration_token,
+            )
+            response = messaging.send_multicast(message)
+            # Response is a message ID string.
+            print('Successfully sent message:', response);
+
+        except exceptions.FirebaseError as error:
+            print('Error sending message:', error);
+        except ValueError as value_error:
+            print('Error sending message:', value_error);
+
+
+    def send_to_token(self, title, body, image_url = None, registration_token = None):
+        if registration_token == None:
+            return "Registration token is null"
+        try: 
+            #notification_image_url = image_url
+            #if image_url == None:
+            notification_image_url = self.logo_url
+            registration_token = 'cOqNt8rzo074gbIkBSpCgW:APA91bEBuNi3iVzGKb4JhxqN2j80MoJbNptLHk2qsdeKBQz5grpHtrPPXvDqn5BJVVSaj1nwGPwgN7pi6FIApog_TTP3g1yobgmgpPN6udrYgzILlVPMvdGGFDSDh6gKlczhlTL9NEp0'
+
+            print(f'image: {notification_image_url}' )    
+
+            message = messaging.Message(
+                notification=messaging.Notification(
+                    title=title,
+                    body=body,
+                ),
+                android=messaging.AndroidConfig(
+                    ttl=datetime.timedelta(seconds=3600),
+                    priority='normal',
+                    notification=messaging.AndroidNotification(
+                        image=notification_image_url,
+                    ),
+                ),
+                apns=messaging.APNSConfig(
+                    payload=messaging.APNSPayload(
+                        aps=messaging.Aps(badge=1),
+                    ),
+                    fcm_options= messaging.APNSFCMOptions(
+                        image=notification_image_url,
+                    )
+                ),
+                token= registration_token,
+            )
+            response = messaging.send(message)
+            # Response is a message ID string.
+            print('Successfully sent message:', response);
+            rc = 'OK'
+
+        except exceptions.FirebaseError as error:
+            print('Error sending message:', error);
+            rc = error
+        except ValueError as value_error:
+            print('Error sending message:', value_error);
+            rc = value_error
+
+        return rc    
+

aitrainer_backoffice/controlling/automation/mautic.py

@@ -6,7 +6,7 @@ import datetime
 from ..models.customer import Customer
 
 
-class MauticHelper:
+class Mautic:
 
     def syncTrial(self):
         tenDays = datetime.datetime - datetime.timedelta(days=10)

aitrainer_backoffice/controlling/automation/messaging.py

@@ -11,27 +11,31 @@ import argparse
 import json
 import requests
 import datetime
+import os
 
 from google.oauth2 import service_account
-from firebase_admin import messaging
+from firebase_admin import messaging, initialize_app
+from aitrainer_backoffice.settings.prod import BASE_DIR
 
 PROJECT_ID = 'aitrainer-af0ec'
 BASE_URL = 'https://fcm.googleapis.com'
 FCM_ENDPOINT = 'v1/projects/' + PROJECT_ID + '/messages:send'
 FCM_URL = BASE_URL + '/' + FCM_ENDPOINT
-SCOPES = ['https://www.googleapis.com/auth/firebase.messaging']
+SCOPES = ['https://www.googleapis.com/auth/firebase.messaging', 'https://www.googleapis.com/auth/cloud-platform', 'email']
+ASSET_ROOT = os.path.join(BASE_DIR, "asset")
+SERVER_KEY = 'AAAA18iNwog:APA91bFo_kDhK4Nd_zHAtyPj6D5CR_amV0aenNn9VPMMVz6j8mgyTCSr3J3IDD_U9duBe1AN35oAiWKp6LmDdJ5n2UQU6unfyUUmsFOSaDlEQZtYl6ZeIQ5XW52Fmkc29Xh62GZ4E-Ic'
 
 # [START retrieve_access_token]
 def _get_access_token():
   """Retrieve a valid access token that can be used to authorize requests.
   :return: Access token.
   """
-  credentials = credentials = service_account.Credentials.from_service_account_file(
-    "asset/aitrainer-firebase-adminsdk.json",
-    scopes=['email'],
-  )
-  access_token_info = credentials.get_access_token()
-  return access_token_info.access_token
+
+  default_app = initialize_app()
+  
+  access_token = default_app.credential.get_access_token()
+  return access_token.access_token
+  #return access_token_info.access_token
 # [END retrieve_access_token]
 
 def _send_fcm_message(fcm_message):
@@ -46,13 +50,16 @@ def _send_fcm_message(fcm_message):
   }
   # [END use_access_token]
   resp = requests.post(FCM_URL, data=json.dumps(fcm_message), headers=headers)
+  return resp
 
+'''
   if resp.status_code == 200:
     print('Message sent to Firebase for delivery, response:')
     print(resp.text)
   else:
     print('Unable to send message to Firebase')
     print(resp.text)
+'''
 
 def _build_common_message():
   """Construct common notifiation message.
@@ -120,6 +127,63 @@ def send_to_token():
   print('Successfully sent message:', response)
   # [END send_to_token]
 
+def send_fcm_to_token():
+  # [START send_to_token]
+  # This registration token comes from the client FCM SDKs.  
+
+  '''
+    headers = {
+          'Content-Type': 'application/json',
+          'Authorization': 'Bearer ' + _get_access_token(),
+        }
+
+    # See documentation on defining a message payload.
+    message = messaging.Message( 
+        data = _build_override_message(),
+        token= registration_token,
+    )
+  '''
+
+  registration_token = ['fFjCZmrHREpRvxZMIKhNSI:APA91bH7cfctHFHbKxtQ5XGRlL26jgLLzo3a1x4hlPfZYi9WxrauMkdIBmqnIQnyD8Jc3xEs0gAsgNYNMLDEgdrHV3bbH4gvFHYUrYzOHZFr-2aVCsYF9otT8_fmAV380egGf5HiCIYd',
+  'cOqNt8rzo074gbIkBSpCgW:APA91bEBuNi3iVzGKb4JhxqN2j80MoJbNptLHk2qsdeKBQz5grpHtrPPXvDqn5BJVVSaj1nwGPwgN7pi6FIApog_TTP3g1yobgmgpPN6udrYgzILlVPMvdGGFDSDh6gKlczhlTL9NEp0',
+  'eUrsvYw9ekx0sr_R8pGTD8:APA91bHJKl1D9gLBz7xi0gILb7ng576DDnCvNba7MHqRaFn9MeeCSVLhEU1yC10b1v0KrZ4pVYgUqxkSv2t-Rh0mXtHR7ABGQENuGDfqjokWPNamXhp99Fuq66o3jnlXxSzRKe_aSCtk']
+  
+  message = messaging.MulticastMessage(
+    notification=messaging.Notification(
+        title='$GOOG up 1.43% on the day',
+        body='$GOOG gained 11.80 points to close at 835.67, up 1.43% on the day.',
+    ),
+    android=messaging.AndroidConfig(
+        ttl=datetime.timedelta(seconds=3600),
+        priority='normal',
+        notification=messaging.AndroidNotification(
+            icon='stock_ticker_update',
+            color='#f45342'
+        ),
+    ),
+    apns=messaging.APNSConfig(
+        payload=messaging.APNSPayload(
+            aps=messaging.Aps(badge=42),
+        ),
+    ),
+    tokens= registration_token,
+  )
+
+  print(f'SEND FCM message  {message}')
+  response = messaging.send_multicast(message)
+
+  #data = json.dumps(message)
+
+  #print(f'SEND FCM message {FCM_URL} - Header {headers} - body: {data}')
+
+  #response = requests.post(FCM_URL, headers = headers, data=data)
+  print(f'RESPONSE: {response}')
+
+  #response = _send_fcm_message(message)
+  # Response is a message ID string.
+  #print('Successfully sent message:', response)
+  # [END send_to_token]  
+
 def send_to_topic():
     # [START send_to_topic]
     # The topic name can be optionally prefixed with "/topics/".
@@ -141,6 +205,7 @@ def send_to_topic():
     # [END send_to_topic]
 
 
+
 def send_to_condition():
     # [START send_to_condition]
     # Define a condition which will send to devices which are subscribed

aitrainer_backoffice/controlling/automation/notification.py

@@ -0,0 +1,44 @@
+import datetime
+from .notification_hook import NotificationHook
+from ..models import notification as notif
+from ..models.notification import NotificationHistory
+from .fcm import FCM
+
+class Notification:
+    fcm = FCM()
+
+    def run(self):  
+        notification_queryset = notif.Notification.objects.using('live').raw('SELECT * from notification WHERE active = 1')
+        
+        for notification in notification_queryset:
+            if notification.schedule_date != None:
+                pass
+            elif notification.schedule_hook != None:
+                hook = NotificationHook()
+                try: 
+                    hook_function = notification.schedule_hook
+                    hook_sql = notification.schedule_sql
+                    if hook_sql == None:
+                        customers = getattr(hook, hook_function)()
+                    else:
+                        customers = getattr(hook, hook_function)(hook_sql)
+
+                    for customer in customers:                        
+                        if customer.firebase_reg_token != None:                            
+                            print(f'-- Notify Customer {customer.customer_id}')
+                            rc= self.fcm.send_to_token(notification.message_title, notification.message_body, notification.image_url, customer.firebase_reg_token)
+                            self.insert_history(notification=notification, customer=customer, rc=rc)
+                            
+                            
+                except Exception as ex: 
+                    print(f'Notification Hook {notification.schedule_hook} has no callback function: {ex}')
+
+    def insert_history(self, notification, customer, rc):
+        history = NotificationHistory()
+        history.pk = None
+        history.notification = notification
+        history.customer = customer
+        history.response = rc
+        history.notification_date = datetime.datetime.now()
+        history.save()
+        print(f'-- Notification History "{history}" has been saved')

aitrainer_backoffice/controlling/automation/notification_hook.py

@@ -0,0 +1,16 @@
+from ..models.customer import Customer
+import datetime
+
+class NotificationHook:
+    def __init__(self) -> None:
+        pass
+
+    def NotificationSelectAdmins(self):
+        print(datetime.datetime.now(), " *** START automation NotificationSelectAdmins ")
+        qs = Customer.objects.raw('SELECT customer_id, firebase_reg_token from customer WHERE admin = 1')
+        return qs
+
+    def NotificationCommonSQL(self, sql):
+        print(datetime.datetime.now(), " *** START automation NotificationCommonSQL ")
+        qs = Customer.objects.raw(sql)
+        return qs    

aitrainer_backoffice/controlling/cron/cron.py

@@ -1,19 +1,27 @@
-from ..mautic import MauticHelper
+from ..automation.notification import Notification
 from django_cron import CronJobBase, Schedule
 import datetime
 
 
 class MyCronJob(CronJobBase):
-    print(datetime.datetime.now(), " *** START sync customers ")
+    
     RUN_EVERY_MINS = 60
-
     schedule = Schedule(run_every_mins=RUN_EVERY_MINS)
     code = 'aitrainer_backoffice.controlling.cron'    # a unique code
-    print(datetime.datetime.now(), " *** END sync customers ")
+    
+    def do(self):
+        print(datetime.datetime.now(), " *** START sync customers ")
+
+
+class NotificationJob(CronJobBase):
+    notif = Notification()
+    RUN_EVERY_MINS = 60
+    schedule = Schedule(run_every_mins=RUN_EVERY_MINS)
+    code = 'aitrainer_backoffice.controlling.notification'    # a unique code
 
     def do(self):
-        pass
-        #helper = MauticHelper()
-        #helper.syncTrial()
+        print(datetime.datetime.now(), " *** START notification ")       
+        self.notif.run()
+        print(datetime.datetime.now(), " *** END notification ")
 
 

aitrainer_backoffice/controlling/mautic/__init__.py

@@ -1 +0,0 @@
-from .helper import MauticHelper

aitrainer_backoffice/controlling/models/__init__.py

@@ -1,4 +1,3 @@
 from .customer import Customer
 from .exercises import Exercises
-from .exercise_type import ExerciseType
 from .frequent_customers import FrequentCustomers

aitrainer_backoffice/controlling/models/customer.py

@@ -1,15 +1,6 @@
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
-
-
-class Sport(models.Model):
-    sport_id = models.AutoField(primary_key = True)
-    language_code = models.CharField(max_length=2,default="hu")
-    sport_name = models.CharField(max_length=100)
-
-    class Meta:
-        db_table = 'sport_translation'
-
+from aitrainer_backoffice.models.sports import Sport
 
 class Customer(models.Model):
     customer_id = models.BigAutoField(primary_key=True)
@@ -21,6 +12,7 @@ class Customer(models.Model):
     fitness_level = models.CharField(max_length=20)
     date_add = models.DateField()
     synced_date = models.DateTimeField(blank=True,null=True)
+    firebase_reg_token = models.CharField(max_length=255, blank=True, null=True)
 
     def has_add_permission(self, request):
         return False

aitrainer_backoffice/controlling/models/exercise_type.py

@@ -1,28 +0,0 @@
-from django.db import models
-from django.utils.translation import ugettext_lazy as _
-
-from ..models.exercises import Exercises
-
-
-class ExerciseType(models.Model):
-    exercise_type_id = models.AutoField(primary_key=True)
-    name = models.CharField(max_length=50)
-    active = models.BooleanField()
-
-
-    def has_add_permission(self, request):
-        return False
-
-    def has_delete_permission(self, request, obj=None):
-        return False
-
-    @property
-    def exercise_count(self):
-        count = Exercises.objects.filter(exercise_type_id=self.exercise_type_id).count()
-        return count
-
-    class Meta:
-        db_table = 'exercise_type'
-        verbose_name = _("Frequent Exercise")
-        verbose_name_plural = _("Frequent Exercises")
-

aitrainer_backoffice/controlling/models/exercises.py

@@ -1,6 +1,4 @@
 from django.db import models
-
-#from ..models.exercise_type import ExerciseType
 from ..models.customer import Customer
 
 

aitrainer_backoffice/controlling/models/notification.py

@@ -0,0 +1,18 @@
+from django.db import models
+
+from aitrainer_backoffice.models.notification import Notification
+from ..models import Customer
+
+class NotificationHistory(models.Model):
+    notification_history_id = models.AutoField(primary_key=True)
+    notification = models.ForeignKey(Notification, on_delete=models.CASCADE)
+    customer = models.ForeignKey(Customer, on_delete=models.CASCADE)    
+    response = models.CharField(max_length=255)
+    notification_date = models.DateField(blank=True, null=True)
+
+    class Meta:
+        db_table = 'notification_history'
+        app_label = 'controlling'
+
+    def __str__(self):
+        return f'{self.notification};{self.customer};{self.response}'