Adding the possibility to load team assignments from the googledoc csv

file.

Author: madhephaestus

LabCodeRepoSetup/.gitignore

@@ -2,4 +2,5 @@
 /bin/
 /.project
 /.classpath
-.settings
+.settings
+/2002.csv

LabCodeRepoSetup/src/main/java/edu/wpi/rbe/LabCodeRepoSetupMain.java

@@ -9,6 +9,7 @@
 import java.io.InputStreamReader;
 import java.lang.reflect.Type;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.Collection;
 import java.util.HashMap;
 import java.util.HashSet;
@@ -50,12 +51,57 @@ public static void main(String[] args) throws Exception {
 		}.getType();
 		String json = FileUtils.readFileToString(new File(teamAssignmentsFile));
 		HashMap<String, ArrayList<String>> teamAssignments = gson.fromJson(json, collectionType);
-
 		String projectDestBaseName = teamAssignments.get("projectName").get(0);
 		ArrayList<String> repoDestBaseNames = teamAssignments.get("repoDestBaseNames");
 		String teamDestBaseName = teamAssignments.get("teamDestBaseName").get(0);
 		numberOfTeams = Integer.parseInt(teamAssignments.get("numberOfTeams").get(0));
 
+		if (args.length == 2) {
+			String csvFileName = args[1];
+			if (csvFileName.toLowerCase().endsWith(".csv")) {
+				File csv = new File(csvFileName);
+				String csvData = FileUtils.readFileToString(csv);
+				if (csv.exists()) {
+					String lines[] = csvData.split("\\r?\\n");
+					int teamNum = 0;
+					ArrayList<String> team = new ArrayList<>();
+					for (String line : lines) {
+						List<String> fields = Arrays.asList(line.split(","));
+						int lastTeamNum = teamNum;
+						try {
+
+							teamNum = Integer.parseInt(fields.get(5));
+
+						} catch (Exception ex) {
+							// System.out.println(fields);
+
+							// ex.printStackTrace();
+						}
+
+						if (teamNum > 0 && teamNum != lastTeamNum) {
+							if (lastTeamNum == numberOfTeams)
+								break;
+							System.out.println("Team # " + teamNum);
+							team = new ArrayList<>();
+							String teamString = teamNum > 9 ? "" + teamNum : "0" + teamNum;
+							teamAssignments.put(teamString, team);
+						}
+						if (teamNum > 0) {
+							try {
+								String username = fields.get(3);
+								System.out.println("\t"+username);
+								team.add(username);
+							}catch(Exception e) {
+								break;// end of the list
+							}
+							
+						}
+					}
+
+				}
+			}
+		}
+
 		GitHub github = GitHub.connect();
 		GHOrganization dest = github.getMyOrganizations().get(projectDestBaseName);
 
@@ -80,8 +126,7 @@ public static void main(String[] args) throws Exception {
 		for (GHUser c : currentMembers) {
 			boolean isTeach = false;
 			for (GHUser t : teachingStaff) {
-				if (t.getLogin().contains(c.getLogin())|| 
-						t.getLogin().contains("madhephaestus")) {
+				if (t.getLogin().contains(c.getLogin()) || t.getLogin().contains("madhephaestus")) {
 					isTeach = true;
 					break;
 				}
@@ -123,7 +168,7 @@ public static void main(String[] args) throws Exception {
 						tmp.mkdirs();
 					}
 					tmp.deleteOnExit();
-					cloneDirString=tmp.getAbsolutePath() + "/" + sourceRepo;
+					cloneDirString = tmp.getAbsolutePath() + "/" + sourceRepo;
 					cloneDir = new File(cloneDirString);
 					if (cloneDir.exists()) {
 						System.out.println(cloneDir.getAbsolutePath() + " Exists");
@@ -172,27 +217,27 @@ public static void main(String[] args) throws Exception {
 				System.out.println("Team Found: " + team.getName());
 				for (String member : members) {
 					try {
-					GHUser memberGH = github.getUser(member);
-					if (memberGH == null) {
-						System.out.println("ERROR GitHub user " + member + " does not exist");
-						continue;
-					}
-					if (!team.hasMember(memberGH)) {
-						try {
-							team.add(memberGH, Role.MAINTAINER);
-							System.out.println("Adding " + member + " to " + team.getName());
-						}catch (Exception e) {
-							System.out.println("Inviting " + member + " to " + team.getName());
+						GHUser memberGH = github.getUser(member);
+						if (memberGH == null) {
+							System.out.println("ERROR GitHub user " + member + " does not exist");
+							continue;
+						}
+						if (!team.hasMember(memberGH)) {
+							try {
+								team.add(memberGH, Role.MAINTAINER);
+								System.out.println("Adding " + member + " to " + team.getName());
+							} catch (Exception e) {
+								System.out.println("Inviting " + member + " to " + team.getName());
 
+							}
 						}
-					}
-					
-					allStudents.add(memberGH);
-					}catch(Exception ex) {
-						System.err.println("\r\n\r\n ERROR "+member+" is not a valid GitHub username\r\n\r\n");
+
+						allStudents.add(memberGH);
+					} catch (Exception ex) {
+						System.err.println("\r\n\r\n ERROR " + member + " is not a valid GitHub username\r\n\r\n");
 					}
 				}
-				
+
 				for (GHUser t : teachingStaff) {
 					if (!t.getLogin().contains("madhephaestus"))
 						team.add(t, Role.MAINTAINER);
@@ -256,7 +301,7 @@ public static void main(String[] args) throws Exception {
 			if (deleteAll)
 				for (GHTeam t : allTeams) {
 					if (t.getName().startsWith("HomeworkTeam")) {
-						System.out.println("Deleting team "+t.getName());
+						System.out.println("Deleting team " + t.getName());
 						t.delete();
 					}
 				}
@@ -268,17 +313,17 @@ public static void main(String[] args) throws Exception {
 				GHRepository repositorie = dest.getRepository(hwRepoName);
 				if (repositorie == null) {
 					repositorie = createRepository(dest, hwRepoName, "Homework for " + u.getLogin());
-					System.out.println("Creating Student Homework team "+hwRepoName);
+					System.out.println("Creating Student Homework team " + hwRepoName);
 				}
 				GHTeam myTeam = existingTeams.get(hwTeam);
 				if (myTeam == null) {
 					myTeam = dest.createTeam(hwTeam, GHOrganization.Permission.ADMIN, repositorie);
-					
+
 				}
 				try {
 					myTeam.add(u, Role.MAINTAINER);
-				}catch(Exception ex) {
-					System.out.println("Inviting "+u.getLogin()+" to "+hwTeam);
+				} catch (Exception ex) {
+					System.out.println("Inviting " + u.getLogin() + " to " + hwTeam);
 				}
 				myTeam.add(repositorie, GHOrganization.Permission.ADMIN);
 				for (GHUser t : teachingStaff) {

LabCodeRepoSetup/teamAssignments2002.json

@@ -3,8 +3,6 @@
 "projectName":["RBE200x-lab"],
 "repoDestBaseNames":["RBE2002Code"],
 "teamDestBaseName":["RBE200xTeam"],
-"numberOfTeams":["21"],
-"RBE2002Code":["WPIRoboticsEngineering","RBE2002_template"],
-"20": ["cbp952"],
-"21": ["cbp952"]
+"numberOfTeams":["14"],
+"RBE2002Code":["WPIRoboticsEngineering","RBE2002_template"]
 }