Use a context manager to save an object on exit












0












$begingroup$


The following code uses a context manager to store and load variables into files.
However, it's very annoying to have to set the value property of what is yielded by the context manager (loaded.value).
I would like to




  1. Not have to define a new class like LoadedValue

  2. Set the yielded value of the context manager (loaded) to be the value that is saved.


Any way of doing either of those two (as well as other helpful improvements) would be greatly appreciated.



import os
import pickle
from contextlib import contextmanager

class LoadedValue:
def __init__(self, value):
self.value = value

def __str__(self):
return "<LoadedValue: {}>".format(self.value)

@contextmanager
def load_manager(load_file="file.pkl"):
with open(load_file, "rb") as f:
loaded_object = LoadedValue(pickle.load(f))
try:
yield loaded_object
finally:
with open(load_file, "wb+") as f:
pickle.dump(loaded_object.value, f)

if __name__ == "__main__":
filename = "test.pkl"
with open(filename, "wb+") as f:
pickle.dump(7, f)

with load_manager(filename) as loaded:
print(loaded)
loaded.value = 5 # this is what I have to do
# loaded = 5 # this is what I want to do

with load_manager(filename) as loaded:
print(loaded) # >> 5








share







New contributor




Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
Check out our Code of Conduct.







$endgroup$

















    0












    $begingroup$


    The following code uses a context manager to store and load variables into files.
    However, it's very annoying to have to set the value property of what is yielded by the context manager (loaded.value).
    I would like to




    1. Not have to define a new class like LoadedValue

    2. Set the yielded value of the context manager (loaded) to be the value that is saved.


    Any way of doing either of those two (as well as other helpful improvements) would be greatly appreciated.



    import os
    import pickle
    from contextlib import contextmanager

    class LoadedValue:
    def __init__(self, value):
    self.value = value

    def __str__(self):
    return "<LoadedValue: {}>".format(self.value)

    @contextmanager
    def load_manager(load_file="file.pkl"):
    with open(load_file, "rb") as f:
    loaded_object = LoadedValue(pickle.load(f))
    try:
    yield loaded_object
    finally:
    with open(load_file, "wb+") as f:
    pickle.dump(loaded_object.value, f)

    if __name__ == "__main__":
    filename = "test.pkl"
    with open(filename, "wb+") as f:
    pickle.dump(7, f)

    with load_manager(filename) as loaded:
    print(loaded)
    loaded.value = 5 # this is what I have to do
    # loaded = 5 # this is what I want to do

    with load_manager(filename) as loaded:
    print(loaded) # >> 5








    share







    New contributor




    Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
    Check out our Code of Conduct.







    $endgroup$















      0












      0








      0





      $begingroup$


      The following code uses a context manager to store and load variables into files.
      However, it's very annoying to have to set the value property of what is yielded by the context manager (loaded.value).
      I would like to




      1. Not have to define a new class like LoadedValue

      2. Set the yielded value of the context manager (loaded) to be the value that is saved.


      Any way of doing either of those two (as well as other helpful improvements) would be greatly appreciated.



      import os
      import pickle
      from contextlib import contextmanager

      class LoadedValue:
      def __init__(self, value):
      self.value = value

      def __str__(self):
      return "<LoadedValue: {}>".format(self.value)

      @contextmanager
      def load_manager(load_file="file.pkl"):
      with open(load_file, "rb") as f:
      loaded_object = LoadedValue(pickle.load(f))
      try:
      yield loaded_object
      finally:
      with open(load_file, "wb+") as f:
      pickle.dump(loaded_object.value, f)

      if __name__ == "__main__":
      filename = "test.pkl"
      with open(filename, "wb+") as f:
      pickle.dump(7, f)

      with load_manager(filename) as loaded:
      print(loaded)
      loaded.value = 5 # this is what I have to do
      # loaded = 5 # this is what I want to do

      with load_manager(filename) as loaded:
      print(loaded) # >> 5








      share







      New contributor




      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.







      $endgroup$




      The following code uses a context manager to store and load variables into files.
      However, it's very annoying to have to set the value property of what is yielded by the context manager (loaded.value).
      I would like to




      1. Not have to define a new class like LoadedValue

      2. Set the yielded value of the context manager (loaded) to be the value that is saved.


      Any way of doing either of those two (as well as other helpful improvements) would be greatly appreciated.



      import os
      import pickle
      from contextlib import contextmanager

      class LoadedValue:
      def __init__(self, value):
      self.value = value

      def __str__(self):
      return "<LoadedValue: {}>".format(self.value)

      @contextmanager
      def load_manager(load_file="file.pkl"):
      with open(load_file, "rb") as f:
      loaded_object = LoadedValue(pickle.load(f))
      try:
      yield loaded_object
      finally:
      with open(load_file, "wb+") as f:
      pickle.dump(loaded_object.value, f)

      if __name__ == "__main__":
      filename = "test.pkl"
      with open(filename, "wb+") as f:
      pickle.dump(7, f)

      with load_manager(filename) as loaded:
      print(loaded)
      loaded.value = 5 # this is what I have to do
      # loaded = 5 # this is what I want to do

      with load_manager(filename) as loaded:
      print(loaded) # >> 5






      python-3.x





      share







      New contributor




      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.










      share







      New contributor




      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.








      share



      share






      New contributor




      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.









      asked 1 min ago









      Pro QPro Q

      101




      101




      New contributor




      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.





      New contributor





      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.






      Pro Q is a new contributor to this site. Take care in asking for clarification, commenting, and answering.
      Check out our Code of Conduct.






















          0






          active

          oldest

          votes











          Your Answer





          StackExchange.ifUsing("editor", function () {
          return StackExchange.using("mathjaxEditing", function () {
          StackExchange.MarkdownEditor.creationCallbacks.add(function (editor, postfix) {
          StackExchange.mathjaxEditing.prepareWmdForMathJax(editor, postfix, [["\$", "\$"]]);
          });
          });
          }, "mathjax-editing");

          StackExchange.ifUsing("editor", function () {
          StackExchange.using("externalEditor", function () {
          StackExchange.using("snippets", function () {
          StackExchange.snippets.init();
          });
          });
          }, "code-snippets");

          StackExchange.ready(function() {
          var channelOptions = {
          tags: "".split(" "),
          id: "196"
          };
          initTagRenderer("".split(" "), "".split(" "), channelOptions);

          StackExchange.using("externalEditor", function() {
          // Have to fire editor after snippets, if snippets enabled
          if (StackExchange.settings.snippets.snippetsEnabled) {
          StackExchange.using("snippets", function() {
          createEditor();
          });
          }
          else {
          createEditor();
          }
          });

          function createEditor() {
          StackExchange.prepareEditor({
          heartbeatType: 'answer',
          autoActivateHeartbeat: false,
          convertImagesToLinks: false,
          noModals: true,
          showLowRepImageUploadWarning: true,
          reputationToPostImages: null,
          bindNavPrevention: true,
          postfix: "",
          imageUploader: {
          brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
          contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
          allowUrls: true
          },
          onDemand: true,
          discardSelector: ".discard-answer"
          ,immediatelyShowMarkdownHelp:true
          });


          }
          });






          Pro Q is a new contributor. Be nice, and check out our Code of Conduct.










          draft saved

          draft discarded


















          StackExchange.ready(
          function () {
          StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fcodereview.stackexchange.com%2fquestions%2f211517%2fuse-a-context-manager-to-save-an-object-on-exit%23new-answer', 'question_page');
          }
          );

          Post as a guest















          Required, but never shown

























          0






          active

          oldest

          votes








          0






          active

          oldest

          votes









          active

          oldest

          votes






          active

          oldest

          votes








          Pro Q is a new contributor. Be nice, and check out our Code of Conduct.










          draft saved

          draft discarded


















          Pro Q is a new contributor. Be nice, and check out our Code of Conduct.













          Pro Q is a new contributor. Be nice, and check out our Code of Conduct.












          Pro Q is a new contributor. Be nice, and check out our Code of Conduct.
















          Thanks for contributing an answer to Code Review Stack Exchange!


          • Please be sure to answer the question. Provide details and share your research!

          But avoid



          • Asking for help, clarification, or responding to other answers.

          • Making statements based on opinion; back them up with references or personal experience.


          Use MathJax to format equations. MathJax reference.


          To learn more, see our tips on writing great answers.




          draft saved


          draft discarded














          StackExchange.ready(
          function () {
          StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fcodereview.stackexchange.com%2fquestions%2f211517%2fuse-a-context-manager-to-save-an-object-on-exit%23new-answer', 'question_page');
          }
          );

          Post as a guest















          Required, but never shown





















































          Required, but never shown














          Required, but never shown












          Required, but never shown







          Required, but never shown

































          Required, but never shown














          Required, but never shown












          Required, but never shown







          Required, but never shown







          Popular posts from this blog

          Morgemoulin

          Scott Moir

          Souastre