Changeset 9362 for main/waeup.kofa/trunk/tools/copy_media_files.py

Timestamp:

19 Oct 2012, 22:59:41 (12 years ago)

Author:

uli

Message:

Update tools to allow new ids in imports. Support for new import col 'old_id'.

File:

• main/waeup.kofa/trunk/tools/copy_media_files.py (modified) (3 diffs)

main/waeup.kofa/trunk/tools/copy_media_files.py

@@ -28 +28 @@
 ## The Ids of new students to search for in old portal.
 NEW_IDS_CSV = "StudentIds.csv"
+
+## The ids of students that will will get a complete new number, not
+## an updated one. Must contain a mapping of old ids to new ids.
+## If no such update should be performed, please set the constant to None.
+ID_MAP_CSV = None
+ID_MAP_CSV = "id_mapping.csv"
 
 ## Permissions to be set on new files/dirs. Set OWNER and/or GROUP to
@@ -79 +85 @@
 
 def copy_file(file_src, file_dst):
-    create_path(os.path.dirname(dst))
+    create_path(os.path.dirname(file_dst))
     print "COPY FILE: %s -> %s" % (file_src, file_dst)
     shutil.copyfile(file_src, file_dst)
@@ -85 +91 @@
     return
 
+def get_new_old_id_mapping():
+    """Returns a dict mapping from _new_ ids to old (SRP) ids.
+
+    The dict is read from ID_MAP_CSV file. If this var is set to
+    ``None`` an empty dict is returned. The ID_MAP_CSV contains only
+    the student ids of those students, for which the standard method
+    (new_id=CHAR+old_id) does not work.
+    """
+    if ID_MAP_CSV is None:
+        return {}
+    if not os.path.isfile(ID_MAP_CSV):
+        raise IOError(
+            "No such file for mapping new to old ids: %s" % ID_MAP_CSV)
+    result = dict()
+    reader = csv.DictReader(open(ID_MAP_CSV, 'rb'))
+    for row in reader:
+        result[row['new_id']] = row['student_id']
+    return result
+
+# special ids not handled in common way
+new_to_old_map = get_new_old_id_mapping()
 removable_dirs = []
 reader = csv.DictReader(open(NEW_IDS_CSV, 'rb'))
 for row in reader:
     new_stud_id = row['student_id']
-    stud_id = new_stud_id[1:]
+    stud_id = new_to_old_map.get(new_stud_id[1:], new_stud_id[1:])
     src_folder = os.path.join(SRC_DIR, stud_id[0], stud_id)
     dst_folder = os.path.join(DST_DIR, new_folder_name(stud_id), new_stud_id)