Add dry run mode

MaxSem · MaxSem · commit b618ede5110c · 2019-05-20T19:02:10.000-07:00
Adding the --dry-run flag will allow to output new page content instead
of posting it to the wiki.

Includes allowing to specify the port in configuration to make testing
possible.
diff --git a/main.py b/main.py
@@ -7,21 +7,31 @@
 
 def main ( args ):
 	wiki = args[1]
+	dry_run = False
+	methods = []
 	for arg in args[2:]:
-		print arg
-		run = Run( wiki )
-		method = getattr( run, str( arg ) )
+		if arg == '--dry-run':
+			dry_run = True
+		else:
+			methods.append(arg)
+
+	for m in methods:
+		print( m )
+		run = Run( wiki, dry_run )
+		method = getattr( run, str( m ) )
 		if not method:
 			raise Exception( "Method not implemented" )
 		else:
 			method()
 
 class Run:
-	def __init__( self, wiki ):
-		self.db = pymysql.connect( host = wiki + 'wiki.labsdb', user = credentials['user'], passwd = credentials['pass'], db = wiki + 'wiki_p' )
+	def __init__( self, wiki, dry_run ):
+		host = '127.0.0.1' if dry_run else wiki + 'wiki.labsdb'
+		port = credentials.get( 'port', 3306 )
+		self.db = pymysql.connect( host = host, user = credentials['user'], passwd = credentials['pass'], db = wiki + 'wiki_p', port = port )
 		self.site = mwclient.Site( wiki + '.wikipedia.org' )
 		self.site.login( cttbot['user'], cttbot['pass'] )
-		self.rep = Reports( self.site, self.db, wiki )
+		self.rep = Reports( self.site, self.db, wiki, dry_run )
 
 	def forgotten_articles( self ):
 		self.rep.forgotten_articles()
diff --git a/reports.py b/reports.py
@@ -9,12 +9,13 @@
 sys.setdefaultencoding('utf8')
 
 class Reports:
-	def __init__( self, site, db, wiki ):
+	def __init__( self, site, db, wiki, dry_run=False ):
 		self.db = db
 		self.site = site
 		self.wiki = wiki
 		# 30s is not enough
 		self.site.requests['timeout'] = 120
+		self.dry_run = dry_run
 
 	# Oldest edited articles
 	# Run time on enwiki 5 hours 23 minutes as of 8 Sept 2015
@@ -446,8 +447,11 @@ def publish_report( self, title, content ):
 		reports_base_url = dict_obj[ str( 'reports_base_url' ) ]
 		report_title = dict_obj[ str( title ) ]
 		print str( reports_base_url + report_title )
-		page = self.site.Pages[ reports_base_url + report_title ]
-		page.save( content, summary = dict_obj[ 'summary' ] , minor=True)
+		if self.dry_run:
+			print content
+		else:
+			page = self.site.Pages[ reports_base_url + report_title ]
+			page.save( content, summary = dict_obj[ 'summary' ] , minor=True)
 
 
 	def linkify( self, title, namespace = None ):
