diff env/lib/python3.7/site-packages/planemo/training/utils.py @ 0:26e78fe6e8c4 draft

"planemo upload commit c699937486c35866861690329de38ec1a5d9f783"
author shellac
date Sat, 02 May 2020 07:14:21 -0400
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.7/site-packages/planemo/training/utils.py	Sat May 02 07:14:21 2020 -0400
@@ -0,0 +1,63 @@
+"""Module contains code for the Requirement, Reference and some general functions for training."""
+
+import collections
+
+import oyaml as yaml
+
+
+class Requirement(object):
+    """Class to describe a training requirement."""
+
+    def __init__(self, req_type="internal", topic_name="introduction", title=None, tutorials=None, link=None):
+        """Init a Requirement instance."""
+        self.type = req_type
+        self.topic_name = topic_name
+        self.tutorials = tutorials
+        self.title = title
+        self.link = link
+
+    def init_from_dict(self, metadata):
+        """Init from a dictionary generated by export_to_ordered_dict."""
+        self.type = metadata['type']
+        if self.type == 'internal':
+            self.topic_name = metadata['topic_name']
+            if 'tutorials' in metadata:
+                self.tutorials = metadata['tutorials']
+        else:
+            self.title = metadata['title']
+            if self.type == 'external':
+                self.link = metadata['link']
+
+    def export_to_ordered_dict(self):
+        """Export the requirement into an ordered dictionary."""
+        req = collections.OrderedDict()
+        req['type'] = self.type
+        if self.type == 'internal':
+            req['topic_name'] = self.topic_name
+            if self.tutorials:
+                req['tutorials'] = self.tutorials
+        else:
+            req['title'] = self.title
+            if self.type == 'external':
+                req['link'] = self.link
+        return req
+
+
+def load_yaml(filepath):
+    """Load the content of a YAML file to a dictionary."""
+    with open(filepath, "r") as m_file:
+        content = yaml.safe_load(m_file)
+    return content
+
+
+def save_to_yaml(content, filepath):
+    """Save a dictionary to a YAML file."""
+    with open(filepath, 'w') as stream:
+        yaml.safe_dump(content,
+                       stream,
+                       indent=2,
+                       default_flow_style=False,
+                       default_style='',
+                       explicit_start=True,
+                       encoding='utf-8',
+                       allow_unicode=True)