feat: add generate_from_queryset

Author: birddevelper

django_query_to_table/DjangoQtt.py

@@ -100,6 +100,15 @@ def generateFromSql(cursor, title, sqltext, footerCols=None, htmlClass="", direc
             return 'Not Valid SQL'
 
       return _generate_report(cursor, title, sqltext, footerCols, htmlClass, direction, font, totalText, rowIndex, headerRowColor, evenRowColor, oddRowColor)
+   except Exception as e:
+      logger.error(traceback.format_exc())
+      return f"Error: {str(e)}"
+   
+def generate_from_queryset(title, queryset, footerCols=None, htmlClass="", direction="ltr", font="Tahoma", totalText="Total", rowIndex=False, headerRowColor='#eeeeee', evenRowColor='#ffffff', oddRowColor="#ffffff"):
+   """Generate the SQL report using a Django QuerySet."""
+   try:
+      sqltext = str(queryset.query)
+      return generate_from_sql(title, sqltext, footerCols, htmlClass, direction, font, totalText, rowIndex, headerRowColor, evenRowColor, oddRowColor)
    except Exception as e:
       logger.error(traceback.format_exc())
       return f"Error: {str(e)}"