first commit

portal_stripe_payment/models/__init__.py

@@ -0,0 +1,4 @@
+from . import models
+from . import subscription
+from . import account_move
+from . import res_partner

portal_stripe_payment/models/account_move.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+from odoo import models, fields, api, _
+import logging
+
+_logger = logging.getLogger(__name__)
+
+
+class AccountMove(models.Model):
+    _inherit = 'account.move'
+    related_stripe_id = fields.Char(string="Related Stripe Invoice")

portal_stripe_payment/models/models.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+import base64
+
+from odoo import models, fields, api, _
+import logging
+
+_logger = logging.getLogger(__name__)
+
+
+class StripeCheckoutSession(models.Model):
+    _name = 'portal.stripe.checkout.session'
+    name = fields.Char()
+    user_id = fields.Many2one('res.users', 'Client User')
+    session_id = fields.Char()
+    session_completed = fields.Boolean(default=False)
+    completion_payload = fields.Text()
+
+    def post_session_completion_tasks(self, session=None):
+        if session is None:
+            session = {}
+            _logger.error('session is None, Exiting')
+
+        session_object = session.get('data', {'object': False}).get('object')
+        if session_object:
+            session_id = session_object.get('id')
+            client_reference_id = session_object.get('client_reference_id')
+            if session_id and client_reference_id:
+                db_session = self.sudo().search([('session_id', '=', session_id), ('session_completed', '=', False),
+                                                 ('user_id', '=', int(client_reference_id))], limit=1)
+                if db_session:  # <<<----------
+                    db_session.sudo().write({'completion_payload': str(session), 'session_completed': True})
+                    stripe_sub_id = session_object.get('subscription')
+                    if stripe_sub_id:
+                        self.env['sale.subscription'].sudo().create_from_stripe_api(user_id=client_reference_id,
+                                                                                    stripe_sub_id=stripe_sub_id)
+                else:
+                    _logger.error('Unable to find stripe checkout session on db')
+            else:
+                _logger.error('Session id is not found, Exiting')
+        else:
+            _logger.error('session object not found, Exiting')
+
+

portal_stripe_payment/models/res_partner.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+from odoo import models, fields, api, _
+import logging
+
+_logger = logging.getLogger(__name__)
+
+
+class Partner(models.Model):
+    _inherit = 'res.partner'
+    related_stripe_id = fields.Char()

portal_stripe_payment/models/subscription.py

@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+import datetime
+
+from odoo import models, fields, api, _
+import stripe
+import logging
+from odoo.exceptions import UserError, ValidationError
+
+_logger = logging.getLogger(__name__)
+
+
+class SaleSubscription(models.Model):
+    _inherit = 'sale.subscription'
+    stripe_subscription_id = fields.Char(string="Related Stripe Subscription")
+
+    def create_from_stripe_api(self, user_id=False, stripe_sub_id=False):
+        if not stripe_sub_id or not user_id:
+            return
+        stripe.api_key = self.env['ir.config_parameter'].sudo().get_param('stripe_secret_api_key')
+        ssub = stripe.Subscription.retrieve(str(stripe_sub_id))
+        if ssub:
+            try:
+                stripe_price_id = ssub.get('items').get('data')[0].get('price').get('id')
+                if stripe_price_id:
+                    saas_package = self.env['saas.package'].sudo().search([('stripe_product_id', '=', stripe_price_id)])
+                    if saas_package and saas_package.subscription_template:
+                        self._create_from_stripe_api(user_id, saas_package, stripe_sub_id)
+                    else:
+                        _logger.error('Unable to find SaaS package or Subscription Template')
+                else:
+                    _logger.error('Unable to find Stripe Price ID')
+
+            except Exception as e:
+                _logger.error(str(e))
+        else:
+            _logger.error('Unable to find Stripe Subscription object')
+
+    def _create_from_stripe_api(self, user_id, package, stripe_sub_id):
+        user = self.env['res.users'].sudo().search([('id', '=', user_id)], limit=1)
+        if user and package and stripe_sub_id:
+            new_sub = self.env['sale.subscription'].sudo().create({
+                'partner_id': user.partner_id.id,
+                'template_id': package.subscription_template.id,
+                'stripe_subscription_id': stripe_sub_id,
+                'recurring_invoice_line_ids':
+                    [(0, 0,
+                      {'product_id': package.year_product_id.id, 'name': package.name, 'price_unit': package.year_price,
+                       'uom_id': package.year_product_id.uom_id.id})]
+            })
+            started = new_sub.sudo().start_subscription()
+            if started:
+                try:
+                    invoice_action = new_sub.sudo().generate_recurring_invoice()
+                    invoice_id = invoice_action.get('res_id')
+                    if invoice_id:
+                        invoice_obj = self.env['account.move'].sudo().browse([invoice_id])
+                        if invoice_obj:
+                            invoice_obj.action_post()
+                            new_sub.sudo().create_saas_app_from_subscription(user_id=user_id, package=package)
+
+                        #     Payment = self.env['account.payment'].with_context(default_line_ids=invoice_obj.invoice_line_ids.ids, default_invoice_ids=[(4, invoice_id, False)])
+                        #     print(Payment)
+                        #     payment_vals = {
+                        #         'date': datetime.date.today(),
+                        #         'amount': invoice_obj.amount_total,
+                        #         'payment_type': 'inbound',
+                        #         'partner_type': 'customer',
+                        #         'partner_id': user.partner_id.id,
+                        #         'line_ids': invoice_obj.invoice_line_ids,
+                        #         # 'ref': self.communication,
+                        #         # 'journal_id': self.journal_id.id,
+                        #         # 'currency_id': self.currency_id.id,
+                        #         # 'partner_bank_id': self.partner_bank_id.id,
+                        #         # 'payment_method_id': self.payment_method_id.id,
+                        #         # 'destination_account_id': self.line_ids[0].account_id.id
+                        #     }
+                        #
+                        #     payment = Payment.sudo().create(payment_vals)
+                        #     print(payment.action_post())
+
+                except UserError as e:
+                    _logger.error(e)
+            else:
+                _logger.error("Unable to start subscription")
+        else:
+            _logger.error('Stripe Subscription Id not found')
+
+    def create_saas_app_from_subscription(self, user_id=False, package=False):
+        saas_app_env = self.env['kk_odoo_saas.app']
+
+        def_vals = saas_app_env.default_get(fields_list=['app_name'])
+
+        if user_id:
+            def_vals['admin_user'] = user_id
+        else:
+            def_vals['admin_user'] = self.partner_id.user_ids.ids[0]
+        app_name = def_vals.get('app_name')
+        configurations = self.env["kk_odoo_saas.k8s.config"]
+        config = configurations.get_default_config()
+        if config:
+            def_vals['configuration'] = config.id
+            def_vals['sub_domain_name'] = app_name
+            def_vals['subscription_id'] = self.id
+            def_vals['client_db_name'] = app_name
+            # def_vals['module_ids'] = [(6, 0, saas_app_ids)]
+            if package:
+                def_vals['docker_image'] = package.docker_image.id
+            def_vals['name'] = '{}'.format(self.code)
+
+            saas_app = saas_app_env.create(def_vals)
+
+            self.build_id = saas_app.id
+
+            _logger.info('Going to Deploy SaaS App, Subscription is going to start')
+            # saas_app.deploy_app()
+        else:
+            _logger.error('Cant create SaaS App, No K8s configuration found')