@@ -214,41 +214,39 @@ def setupDKIM(virtualHostName):
214214
215215 import tldextract
216216
217- #extractDomain = tldextract.extract(virtualHostName)
218- #virtualHostName = extractDomain.domain + '.' + extractDomain.suffix
217+ actualDomain = virtualHostName
218+ extractDomain = tldextract .extract (virtualHostName )
219+ virtualHostName = extractDomain .domain + '.' + extractDomain .suffix
219220
220- if os .path .exists ("/etc/opendkim/keys/" + virtualHostName + "/default.txt" ):
221- return 1 , "None"
221+ if not os .path .exists ("/etc/opendkim/keys/" + virtualHostName + "/default.txt" ):
222+ path = '/etc/opendkim/keys/%s' % (virtualHostName )
223+ command = 'mkdir %s' % (path )
224+ ProcessUtilities .normalExecutioner (command )
222225
226+ ## Generate keys
223227
224- path = '/etc/opendkim/keys/%s' % (virtualHostName )
225- command = 'mkdir %s' % (path )
226- ProcessUtilities .normalExecutioner (command )
227-
228- ## Generate keys
229-
230- if ProcessUtilities .decideDistro () == ProcessUtilities .centos :
231- command = "/usr/sbin/opendkim-genkey -D /etc/opendkim/keys/%s -d %s -s default" % (virtualHostName , virtualHostName )
232- else :
233- command = "opendkim-genkey -D /etc/opendkim/keys/%s -d %s -s default" % (
234- virtualHostName , virtualHostName )
235- ProcessUtilities .normalExecutioner (command )
236- ## Fix permissions
228+ if ProcessUtilities .decideDistro () == ProcessUtilities .centos :
229+ command = "/usr/sbin/opendkim-genkey -D /etc/opendkim/keys/%s -d %s -s default" % (virtualHostName , virtualHostName )
230+ else :
231+ command = "opendkim-genkey -D /etc/opendkim/keys/%s -d %s -s default" % (
232+ virtualHostName , virtualHostName )
233+ ProcessUtilities .normalExecutioner (command )
234+ ## Fix permissions
237235
238- command = "chown -R root:opendkim /etc/opendkim/keys/" + virtualHostName
239- ProcessUtilities .normalExecutioner (command )
236+ command = "chown -R root:opendkim /etc/opendkim/keys/" + virtualHostName
237+ ProcessUtilities .normalExecutioner (command )
240238
241- command = "chmod 640 /etc/opendkim/keys/" + virtualHostName + "/default.private"
242- ProcessUtilities .normalExecutioner (command )
239+ command = "chmod 640 /etc/opendkim/keys/" + virtualHostName + "/default.private"
240+ ProcessUtilities .normalExecutioner (command )
243241
244- command = "chmod 644 /etc/opendkim/keys/" + virtualHostName + "/default.txt"
245- ProcessUtilities .normalExecutioner (command )
242+ command = "chmod 644 /etc/opendkim/keys/" + virtualHostName + "/default.txt"
243+ ProcessUtilities .normalExecutioner (command )
246244
247245 ## Edit key file
248246
249247
250248 keyTable = "/etc/opendkim/KeyTable"
251- configToWrite = "default._domainkey." + virtualHostName + " " + virtualHostName + ":default:/etc/opendkim/keys/" + virtualHostName + "/default.private\n "
249+ configToWrite = "default._domainkey." + actualDomain + " " + actualDomain + ":default:/etc/opendkim/keys/" + virtualHostName + "/default.private\n "
252250
253251 writeToFile = open (keyTable , 'a' )
254252 writeToFile .write (configToWrite )
@@ -257,7 +255,7 @@ def setupDKIM(virtualHostName):
257255 ## Edit signing table
258256
259257 signingTable = "/etc/opendkim/SigningTable"
260- configToWrite = "*@" + virtualHostName + " default._domainkey." + virtualHostName + "\n "
258+ configToWrite = "*@" + actualDomain + " default._domainkey." + actualDomain + "\n "
261259
262260 writeToFile = open (signingTable , 'a' )
263261 writeToFile .write (configToWrite )
@@ -266,7 +264,7 @@ def setupDKIM(virtualHostName):
266264 ## Trusted hosts
267265
268266 trustedHosts = "/etc/opendkim/TrustedHosts"
269- configToWrite = virtualHostName + "\n "
267+ configToWrite = actualDomain + "\n "
270268
271269 writeToFile = open (trustedHosts , 'a' )
272270 writeToFile .write (configToWrite )
0 commit comments