Make the paths constants.

1 files changed, 9 insertions, 5 deletions

diff --git a/ec2-set-apt-sources.py b/ec2-set-apt-sources.py
index 39f25ee6..8bc89312 100755
--- a/ec2-set-apt-sources.py
+++ b/ec2-set-apt-sources.py
@@ -24,24 +24,28 @@ from Cheetah.Template import Template
 
 import ec2init
 
+SOURCES_LIST = '/etc/apt/sources.list'
+GENERATED_SOURCES_LIST = '/var/run/ec2/sources.list'
+OUT_OF_THE_WAY_SOURCES_LIST = '/etc/apt/sources.list.moved-by-ec2-init'
+
 def main():
     ec2 = ec2init.EC2Init()
 
 	mirror = ec2.get_mirror_for_availability_zone()
 
-	if not os.path.exists("/var/run/ec2/sources.lists"):
+	if not os.path.exists(GENERATED_SOURCES_LIST):
 		t = os.popen("lsb_release -cs").read()
 		codename = t.strip()
 
 		mp = { 'mirror' : mirror, 'codename' : codename }
 		t = Template(file='/etc/ec2-init/templates/sources.list.tmpl', searchList=[mp])
-		f = open("/var/run/ec2/sources.list", "w")
+		f = open(GENERATED_SOURCES_LIST, 'w')
 		f.write(t.respond())
 		f.close()
 
-	if not os.path.exists("/etc/apt/sources.list-ec2-init"):
-		os.rename('/etc/apt/sources.list', '/etc/apt/sources.list-ec2-init')
-		os.symlink('/var/run/ec2/sources.list', '/etc/apt/sources.list')
+	if not os.path.exists(OUT_OF_THE_WAY_SOURCES_LIST):
+		os.rename(SOURCES_LIST, OUT_OF_THE_WAY_SOURCES_LIST)
+		os.symlink(GENERATED_SOURCES_LIST, SOURCES_LIST)
         aptget = subprocess.Popen(['apt-get', 'update'])
         aptget.communicate()