Generate enums for Assets.xcassets

generate_image_assets.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-
#  generate_image_assets.py
#  punchcards
#
#  Created by Privat on 13/07/2016.
#  Copyright © 2016 Eazy IT. All rights reserved.
import re
import os

IMAGES_PATH = "./Assets.xcassets"
IMAGES_FILE_PATH = "../Classes/Ressources/Images.swift"

print "Path to image directory ",IMAGES_PATH
print "Path to Swift file ",IMAGES_FILE_PATH

with open(IMAGES_FILE_PATH, "w") as f:

	f.write("//\n\
// Created by Privat on 13/07/2016.\n\
// Copyright (c) 2016 Eazy IT. All rights reserved.\n\
//\n\
\n\
import Foundation\n\
import UIKit\n\
\n\
enum Images: String {\n\n")


	for root, dirs, files in os.walk(IMAGES_PATH):
		for dir in dirs:
			if dir.endswith(".imageset"):
				string = dir.replace(".imageset","")
				string2 = "    case " + string + "\n"
				f.write(string2)


	f.write("\n\
	var image: UIImage? {\n\
		    return UIImage(named: self.rawValue)\n\
	}\n\
}")
f.close()