lib/write_xlsx/package/app.rb in write_xlsx-1.07.0 vs lib/write_xlsx/package/app.rb in write_xlsx-1.08.0

@@ -5,17 +5,19 @@
 module Writexlsx
   module Package
     class App
 
       include Writexlsx::Utility
+      attr_writer :doc_security
 
       def initialize(workbook)
         @writer = Package::XMLWriterSimple.new
         @workbook      = workbook
         @part_names    = []
         @heading_pairs = []
         @properties    = {}
+        @doc_security  = 0
       end
 
       def set_xml_writer(filename)
         @writer.set_xml_writer(filename)
       end
@@ -121,12 +123,10 @@
 
       #
       # Write the <DocSecurity> element.
       #
       def write_doc_security
-        data = 0
-
-        @writer.data_element('DocSecurity', data)
+        @writer.data_element('DocSecurity', @doc_security)
       end
 
       #
       # Write the <ScaleCrop> element.
       #