You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

301 lines
11 KiB

  1. # Author: Julien Coux
  2. # Copyright 2016 Camptocamp SA
  3. # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
  4. from odoo import models
  5. class AbstractReportXslx(models.AbstractModel):
  6. _name = 'report.account_financial_report.abstract_report_xlsx'
  7. _inherit = 'report.report_xlsx.abstract'
  8. def __init__(self, pool, cr):
  9. # main sheet which will contains report
  10. self.sheet = None
  11. # columns of the report
  12. self.columns = None
  13. # row_pos must be incremented at each writing lines
  14. self.row_pos = None
  15. # Formats
  16. self.format_right = None
  17. self.format_right_bold_italic = None
  18. self.format_bold = None
  19. self.format_header_left = None
  20. self.format_header_center = None
  21. self.format_header_right = None
  22. self.format_header_amount = None
  23. self.format_amount = None
  24. self.format_percent_bold_italic = None
  25. def get_workbook_options(self):
  26. return {'constant_memory': True}
  27. def generate_xlsx_report(self, workbook, data, objects):
  28. report = objects
  29. self.row_pos = 0
  30. self._define_formats(workbook)
  31. report_name = self._get_report_name()
  32. filters = self._get_report_filters(report)
  33. self.columns = self._get_report_columns(report)
  34. self.sheet = workbook.add_worksheet(report_name[:31])
  35. self._set_column_width()
  36. self._write_report_title(report_name)
  37. self._write_filters(filters)
  38. self._generate_report_content(workbook, report)
  39. def _define_formats(self, workbook):
  40. """ Add cell formats to current workbook.
  41. Those formats can be used on all cell.
  42. Available formats are :
  43. * format_bold
  44. * format_right
  45. * format_right_bold_italic
  46. * format_header_left
  47. * format_header_center
  48. * format_header_right
  49. * format_header_amount
  50. * format_amount
  51. * format_percent_bold_italic
  52. """
  53. self.format_bold = workbook.add_format({'bold': True})
  54. self.format_right = workbook.add_format({'align': 'right'})
  55. self.format_right_bold_italic = workbook.add_format(
  56. {'align': 'right', 'bold': True, 'italic': True}
  57. )
  58. self.format_header_left = workbook.add_format(
  59. {'bold': True,
  60. 'border': True,
  61. 'bg_color': '#FFFFCC'})
  62. self.format_header_center = workbook.add_format(
  63. {'bold': True,
  64. 'align': 'center',
  65. 'border': True,
  66. 'bg_color': '#FFFFCC'})
  67. self.format_header_right = workbook.add_format(
  68. {'bold': True,
  69. 'align': 'right',
  70. 'border': True,
  71. 'bg_color': '#FFFFCC'})
  72. self.format_header_amount = workbook.add_format(
  73. {'bold': True,
  74. 'border': True,
  75. 'bg_color': '#FFFFCC'})
  76. self.format_header_amount.set_num_format('#,##0.00')
  77. self.format_amount = workbook.add_format()
  78. self.format_amount.set_num_format('#,##0.00')
  79. self.format_percent_bold_italic = workbook.add_format(
  80. {'bold': True, 'italic': True}
  81. )
  82. self.format_percent_bold_italic.set_num_format('#,##0.00%')
  83. def _set_column_width(self):
  84. """Set width for all defined columns.
  85. Columns are defined with `_get_report_columns` method.
  86. """
  87. for position, column in self.columns.items():
  88. self.sheet.set_column(position, position, column['width'])
  89. def _write_report_title(self, title):
  90. """Write report title on current line using all defined columns width.
  91. Columns are defined with `_get_report_columns` method.
  92. """
  93. self.sheet.merge_range(
  94. self.row_pos, 0, self.row_pos, len(self.columns) - 1,
  95. title, self.format_bold
  96. )
  97. self.row_pos += 3
  98. def _write_filters(self, filters):
  99. """Write one line per filters on starting on current line.
  100. Columns number for filter name is defined
  101. with `_get_col_count_filter_name` method.
  102. Columns number for filter value is define
  103. with `_get_col_count_filter_value` method.
  104. """
  105. col_name = 1
  106. col_count_filter_name = self._get_col_count_filter_name()
  107. col_count_filter_value = self._get_col_count_filter_value()
  108. col_value = col_name + col_count_filter_name + 1
  109. for title, value in filters:
  110. self.sheet.merge_range(
  111. self.row_pos, col_name,
  112. self.row_pos, col_name + col_count_filter_name - 1,
  113. title, self.format_header_left)
  114. self.sheet.merge_range(
  115. self.row_pos, col_value,
  116. self.row_pos, col_value + col_count_filter_value - 1,
  117. value)
  118. self.row_pos += 1
  119. self.row_pos += 2
  120. def write_array_title(self, title):
  121. """Write array title on current line using all defined columns width.
  122. Columns are defined with `_get_report_columns` method.
  123. """
  124. self.sheet.merge_range(
  125. self.row_pos, 0, self.row_pos, len(self.columns) - 1,
  126. title, self.format_bold
  127. )
  128. self.row_pos += 1
  129. def write_array_header(self):
  130. """Write array header on current line using all defined columns name.
  131. Columns are defined with `_get_report_columns` method.
  132. """
  133. for col_pos, column in self.columns.items():
  134. self.sheet.write(self.row_pos, col_pos, column['header'],
  135. self.format_header_center)
  136. self.row_pos += 1
  137. def write_line(self, line_object):
  138. """Write a line on current line using all defined columns field name.
  139. Columns are defined with `_get_report_columns` method.
  140. """
  141. for col_pos, column in self.columns.items():
  142. value = getattr(line_object, column['field'])
  143. cell_type = column.get('type', 'string')
  144. if cell_type == 'string':
  145. self.sheet.write_string(self.row_pos, col_pos, value or '')
  146. elif cell_type == 'amount':
  147. self.sheet.write_number(
  148. self.row_pos, col_pos, float(value), self.format_amount
  149. )
  150. self.row_pos += 1
  151. def write_initial_balance(self, my_object, label):
  152. """Write a specific initial balance line on current line
  153. using defined columns field_initial_balance name.
  154. Columns are defined with `_get_report_columns` method.
  155. """
  156. col_pos_label = self._get_col_pos_initial_balance_label()
  157. self.sheet.write(self.row_pos, col_pos_label, label, self.format_right)
  158. for col_pos, column in self.columns.items():
  159. if column.get('field_initial_balance'):
  160. value = getattr(my_object, column['field_initial_balance'])
  161. cell_type = column.get('type', 'string')
  162. if cell_type == 'string':
  163. self.sheet.write_string(self.row_pos, col_pos, value or '')
  164. elif cell_type == 'amount':
  165. self.sheet.write_number(
  166. self.row_pos, col_pos, float(value), self.format_amount
  167. )
  168. self.row_pos += 1
  169. def write_ending_balance(self, my_object, name, label):
  170. """Write a specific ending balance line on current line
  171. using defined columns field_final_balance name.
  172. Columns are defined with `_get_report_columns` method.
  173. """
  174. for i in range(0, len(self.columns)):
  175. self.sheet.write(self.row_pos, i, '', self.format_header_right)
  176. row_count_name = self._get_col_count_final_balance_name()
  177. col_pos_label = self._get_col_pos_final_balance_label()
  178. self.sheet.merge_range(
  179. self.row_pos, 0, self.row_pos, row_count_name - 1, name,
  180. self.format_header_left
  181. )
  182. self.sheet.write(self.row_pos, col_pos_label, label,
  183. self.format_header_right)
  184. for col_pos, column in self.columns.items():
  185. if column.get('field_final_balance'):
  186. value = getattr(my_object, column['field_final_balance'])
  187. cell_type = column.get('type', 'string')
  188. if cell_type == 'string':
  189. self.sheet.write_string(self.row_pos, col_pos, value or '',
  190. self.format_header_right)
  191. elif cell_type == 'amount':
  192. self.sheet.write_number(
  193. self.row_pos, col_pos, float(value),
  194. self.format_header_amount
  195. )
  196. self.row_pos += 1
  197. def _generate_report_content(self, workbook, report):
  198. pass
  199. def _get_report_name(self):
  200. """
  201. Allow to define the report name.
  202. Report name will be used as sheet name and as report title.
  203. :return: the report name
  204. """
  205. raise NotImplementedError()
  206. def _get_report_columns(self, report):
  207. """
  208. Allow to define the report columns
  209. which will be used to generate report.
  210. :return: the report columns as dict
  211. :Example:
  212. {
  213. 0: {'header': 'Simple column',
  214. 'field': 'field_name_on_my_object',
  215. 'width': 11},
  216. 1: {'header': 'Amount column',
  217. 'field': 'field_name_on_my_object',
  218. 'type': 'amount',
  219. 'width': 14},
  220. }
  221. """
  222. raise NotImplementedError()
  223. def _get_report_filters(self, report):
  224. """
  225. :return: the report filters as list
  226. :Example:
  227. [
  228. ['first_filter_name', 'first_filter_value'],
  229. ['second_filter_name', 'second_filter_value']
  230. ]
  231. """
  232. raise NotImplementedError()
  233. def _get_col_count_filter_name(self):
  234. """
  235. :return: the columns number used for filter names.
  236. """
  237. raise NotImplementedError()
  238. def _get_col_count_filter_value(self):
  239. """
  240. :return: the columns number used for filter values.
  241. """
  242. raise NotImplementedError()
  243. def _get_col_pos_initial_balance_label(self):
  244. """
  245. :return: the columns position used for initial balance label.
  246. """
  247. raise NotImplementedError()
  248. def _get_col_count_final_balance_name(self):
  249. """
  250. :return: the columns number used for final balance name.
  251. """
  252. raise NotImplementedError()
  253. def _get_col_pos_final_balance_label(self):
  254. """
  255. :return: the columns position used for final balance label.
  256. """
  257. raise NotImplementedError()