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.

222 lines
7.3 KiB

  1. # -*- coding: utf-8 -*-
  2. # Author: Julien Coux
  3. # Copyright 2016 Camptocamp SA
  4. # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
  5. from openerp.addons.report_xlsx.report.report_xlsx import ReportXlsx
  6. class AbstractReportXslx(ReportXlsx):
  7. def __init__(self, name, table, rml=False, parser=False, header=True,
  8. store=False):
  9. super(AbstractReportXslx, self).__init__(
  10. name, table, rml, parser, header, store)
  11. # main sheet which will contains report
  12. self.sheet = None
  13. # columns of the report
  14. self.columns = None
  15. # row_pos must be incremented at each writing lines
  16. self.row_pos = None
  17. # Formats
  18. self.format_right = None
  19. self.format_bold = None
  20. self.format_header_left = None
  21. self.format_header_center = None
  22. self.format_header_right = None
  23. self.format_header_amount = None
  24. self.format_amount = None
  25. def generate_xlsx_report(self, workbook, data, objects):
  26. report = objects
  27. self.row_pos = 0
  28. self._define_formats(workbook)
  29. report_name = self._get_report_name()
  30. filters = self._get_report_filters(report)
  31. self.columns = self._get_report_columns()
  32. self.sheet = workbook.add_worksheet(report_name[:31])
  33. self._set_column_width()
  34. self._write_report_title(report_name)
  35. self._write_filters(filters)
  36. self._generate_report_content(workbook, report)
  37. def _define_formats(self, workbook):
  38. """ Add cell formats to current workbook.
  39. Those formats can be used on all cell.
  40. Available formats are :
  41. * format_bold
  42. * format_right
  43. * format_header_left
  44. * format_header_center
  45. * format_header_right
  46. * format_header_amount
  47. * format_amount
  48. """
  49. self.format_bold = workbook.add_format({'bold': True})
  50. self.format_right = workbook.add_format({'align': 'right'})
  51. self.format_header_left = workbook.add_format(
  52. {'bold': True,
  53. 'border': True,
  54. 'bg_color': '#FFFFCC'})
  55. self.format_header_center = workbook.add_format(
  56. {'bold': True,
  57. 'align': 'center',
  58. 'border': True,
  59. 'bg_color': '#FFFFCC'})
  60. self.format_header_right = workbook.add_format(
  61. {'bold': True,
  62. 'align': 'right',
  63. 'border': True,
  64. 'bg_color': '#FFFFCC'})
  65. self.format_header_amount = workbook.add_format(
  66. {'bold': True,
  67. 'border': True,
  68. 'bg_color': '#FFFFCC'})
  69. self.format_header_amount.set_num_format('#,##0.00')
  70. self.format_amount = workbook.add_format()
  71. self.format_amount.set_num_format('#,##0.00')
  72. def _set_column_width(self):
  73. """Set width for all defined columns.
  74. Columns are defined with `_get_report_columns` method.
  75. """
  76. for position, column in self.columns.iteritems():
  77. self.sheet.set_column(position, position, column['width'])
  78. def _write_report_title(self, title):
  79. """Write report title on current line using all defined columns width.
  80. Columns are defined with `_get_report_columns` method.
  81. """
  82. self.sheet.merge_range(
  83. self.row_pos, 0, self.row_pos, len(self.columns) - 1,
  84. title, self.format_bold
  85. )
  86. self.row_pos += 3
  87. def _write_filters(self, filters):
  88. """Write one line per filters on starting on current line.
  89. Columns number for filter name is defined
  90. with `_get_col_count_filter_name` method.
  91. Columns number for filter value is define
  92. with `_get_col_count_filter_value` method.
  93. """
  94. col_name = 1
  95. col_count_filter_name = self._get_col_count_filter_name()
  96. col_count_filter_value = self._get_col_count_filter_value()
  97. col_value = col_name + col_count_filter_name + 1
  98. for title, value in filters:
  99. self.sheet.merge_range(
  100. self.row_pos, col_name,
  101. self.row_pos, col_name + col_count_filter_name - 1,
  102. title, self.format_header_left)
  103. self.sheet.merge_range(
  104. self.row_pos, col_value,
  105. self.row_pos, col_value + col_count_filter_value - 1,
  106. value)
  107. self.row_pos += 1
  108. self.row_pos += 2
  109. def write_array_title(self, title):
  110. """Write array title on current line using all defined columns width.
  111. Columns are defined with `_get_report_columns` method.
  112. """
  113. self.sheet.merge_range(
  114. self.row_pos, 0, self.row_pos, len(self.columns) - 1,
  115. title, self.format_bold
  116. )
  117. self.row_pos += 1
  118. def write_array_header(self):
  119. """Write array header on current line using all defined columns name.
  120. Columns are defined with `_get_report_columns` method.
  121. """
  122. for col_pos, column in self.columns.iteritems():
  123. self.sheet.write(self.row_pos, col_pos, column['header'],
  124. self.format_header_center)
  125. self.row_pos += 1
  126. def write_line(self, line_object):
  127. """Write a line on current line using all defined columns field name.
  128. Columns are defined with `_get_report_columns` method.
  129. """
  130. for col_pos, column in self.columns.iteritems():
  131. value = getattr(line_object, column['field'])
  132. cell_type = column.get('type', 'string')
  133. if cell_type == 'string':
  134. self.sheet.write_string(self.row_pos, col_pos, value or '')
  135. elif cell_type == 'amount':
  136. self.sheet.write_number(
  137. self.row_pos, col_pos, float(value), self.format_amount
  138. )
  139. self.row_pos += 1
  140. def _generate_report_content(self, workbook, report):
  141. pass
  142. def _get_report_name(self):
  143. """
  144. Allow to define the report name.
  145. Report name will be used as sheet name and as report title.
  146. :return: the report name
  147. """
  148. raise NotImplementedError()
  149. def _get_report_columns(self):
  150. """
  151. Allow to define the report columns
  152. which will be used to generate report.
  153. :return: the report columns as dict
  154. :Example:
  155. {
  156. 0: {'header': 'Simple column',
  157. 'field': 'field_name_on_my_object',
  158. 'width': 11},
  159. 1: {'header': 'Amount column',
  160. 'field': 'field_name_on_my_object',
  161. 'type': 'amount',
  162. 'width': 14},
  163. }
  164. """
  165. raise NotImplementedError()
  166. def _get_report_filters(self, report):
  167. """
  168. :return: the report filters as list
  169. :Example:
  170. [
  171. ['first_filter_name', 'first_filter_value'],
  172. ['second_filter_name', 'second_filter_value']
  173. ]
  174. """
  175. raise NotImplementedError()
  176. def _get_col_count_filter_name(self):
  177. """
  178. :return: the columns number used for filter names.
  179. """
  180. raise NotImplementedError()
  181. def _get_col_count_filter_value(self):
  182. """
  183. :return: the columns number used for filter values.
  184. """
  185. raise NotImplementedError()