fix feature

1 files changed, 28 insertions, 1 deletions

diff --git a/indoteknik_api/models/rest_api.py b/indoteknik_api/models/rest_api.py
index 0a15aad1..65119b52 100644
--- a/indoteknik_api/models/rest_api.py
+++ b/indoteknik_api/models/rest_api.py
@@ -1,7 +1,9 @@
 from odoo import models
+from odoo.http import request
 import datetime
 from pytz import timezone
 import hashlib
+import base64
 
 
 class RestApi(models.TransientModel):
@@ -17,4 +19,29 @@ class RestApi(models.TransientModel):
         return hashlib.md5((salt + '$' + str(value)).encode()).hexdigest()
 
     def md5_salt_valid(self, value, salt, token):
-        return hashlib.md5((salt + '$' + str(value)).encode()).hexdigest() == token
\ No newline at end of file
+        return hashlib.md5((salt + '$' + str(value)).encode()).hexdigest() == token
+
+    def get_single_attachment(self, model, field, id):
+        domain = [
+            ('res_model', '=', model),
+            ('res_field', '=', field),
+            ('res_id', '=', id),
+        ]
+        fields = ['datas', 'mimetype']
+        result = self.env['ir.attachment'].sudo().search_read(domain, fields)
+        return result[0] if len(result) > 0 else None
+
+    def response_attachment(self, data = {}):
+        decode_content = data.get('decode_content', False)
+        if decode_content:
+            data['content'] = base64.b64decode(data['content'])
+
+        return request.make_response(
+            data['content'],
+            [
+                ('Content-Type', data['mimetype']),
+                ('Content-Disposition', 'attachment; filename=%s' % data['filename']),
+                ('Content-Length', len(data['content']))
+            ]
+        )
+    
\ No newline at end of file