commit 21f882213c069fbd579062ad69c97c48e5005a2e
author Chip Boling <chip.boling@adtran.com> Tue Jul 17 13:25:11 2018 -0500
committer Donald Newton <dn8474@att.com> Thu Jul 19 17:34:10 2018 +0000
tree 65167d23305c5a6ef9a8083580297e247afe9768
parent 805c3000d90fdd4fc96b3c9668f51aaad079e0ed

VOL-712: Reboot OMCI Task & Create/Set Tasks for Enable, Disable support

Change-Id: I95253b0f5b6f1c5080ddc36b02ec61e2414051f9

voltha/extensions/omci/me_frame.py

diff --git a/voltha/extensions/omci/me_frame.py b/voltha/extensions/omci/me_frame.py
index c713bc2..7ce812f 100644
--- a/voltha/extensions/omci/me_frame.py
+++ b/voltha/extensions/omci/me_frame.py
@@ -222,19 +222,21 @@
                 attributes_mask=self.entity_class.mask_for(*mask_set)
             ))
 
-    def reboot(self):
+    def reboot(self, reboot_code=0):
         """
         Create a Reboot request from for this ME
         :return: (OmciFrame) OMCI Frame
         """
         self._check_operation(OP.Reboot)
+        assert 0 <= reboot_code <= 2, 'Reboot code must be 0..2'
 
         return OmciFrame(
             transaction_id=None,
             message_type=OmciReboot.message_id,
             omci_message=OmciReboot(
                 entity_class=getattr(self.entity_class, 'class_id'),
-                entity_id=getattr(self, 'entity_id')
+                entity_id=getattr(self, 'entity_id'),
+                reboot_code=reboot_code
             ))
 
     def mib_reset(self):