|  | @@ -0,0 +1,24 @@
 | 
	
		
			
				|  |  | +#!/usr/bin/env python
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +import yaml
 | 
	
		
			
				|  |  | +import argparse
 | 
	
		
			
				|  |  | +import datetime
 | 
	
		
			
				|  |  | +import csv
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +argp = argparse.ArgumentParser(description='Convert cloc yaml to bigquery csv')
 | 
	
		
			
				|  |  | +argp.add_argument('-i', '--input', type=str)
 | 
	
		
			
				|  |  | +argp.add_argument('-d', '--date', type=str, default=datetime.date.today().strftime('%Y-%m-%d'))
 | 
	
		
			
				|  |  | +argp.add_argument('-o', '--output', type=str, default='out.csv')
 | 
	
		
			
				|  |  | +args = argp.parse_args()
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +data = yaml.load(open(args.input).read())
 | 
	
		
			
				|  |  | +with open(args.output, 'w') as outf:
 | 
	
		
			
				|  |  | +  writer = csv.DictWriter(outf, ['date', 'name', 'language', 'code', 'comment', 'blank'])
 | 
	
		
			
				|  |  | +  for key, value in data.iteritems():
 | 
	
		
			
				|  |  | +    if key == 'header': continue
 | 
	
		
			
				|  |  | +    if key == 'SUM': continue
 | 
	
		
			
				|  |  | +    if key.startswith('third_party/'): continue
 | 
	
		
			
				|  |  | +    row = {'name': key, 'date': args.date}
 | 
	
		
			
				|  |  | +    row.update(value)
 | 
	
		
			
				|  |  | +    writer.writerow(row)
 | 
	
		
			
				|  |  | +
 |