-
Notifications
You must be signed in to change notification settings - Fork 0
/
createreadme.py
103 lines (86 loc) · 3.32 KB
/
createreadme.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from pathlib import Path
HEADER = '''# Competive-Coding
This repository keeps a track of all the competitive problems solved by me.
'''
# GET THE LIST OF ALL THE CATEGORIES
def get_dirs ():
dirs = [x for x in os.listdir ('.') if os.path.isdir (x) and '.git' not in x]
return dirs
#GET PROBLEM STATEMENT
def get_problem (srcfile):
with open (srcfile) as file:
for line in file:
line = line.strip ()
if line.startswith ('//') or line.startswith ('#'):
return line.lstrip ('//').lstrip('#').lstrip()
#FOR EACH DIRECTORY GET THE LINK OF PROGRAMS IN IT
def get_program (category):
program = [x for x in os.listdir (category)]
titles = []
for filename in program:
fullname = os.path.join (category, filename)
if (os.path.isfile (fullname)) and (fullname.endswith ('.cpp') or fullname.endswith ('.c') or fullname.endswith ('.java') or fullname.endswith ('.py')):
problem = get_problem(fullname)
title = Path (fullname).stem
# changing path separator for Windows paths
# https://mail.python.org/pipermail/tutor/2011-July/084788.html
titles.append ((title,problem,'https://github.com/Pranav-Khurana/Competitive-Coding/blob/master/' + fullname.replace (os.path.sep, '/')))
return titles
def get_category_dict (category_names):
categories = {}
count = 0
for category in category_names:
titles = get_program (category)
categories[category] = titles
count += len (titles)
return (count, categories)
#Now we have all the information, print it out in markdown format.
def print_file (category_names, count, categories):
with open ('README.md', 'w') as file:
file.write (HEADER)
if count == 1:
file.write ('>_{0} Problem solved till date_'.format(count))
else:
file.write ('>_{0} Problems solved till date_'.format(count))
file.write ('''
---
## Categories
''')
# print the list of categories with links
for category in sorted (category_names):
file.write ('* [{0}](#{1})\n'.format (category,category.lower()))
if len (category_names) > 0:
file.write ('''
---
''')
# print the section for each category
for category in sorted (category_names):
file.write ('### {0}\n'.format (category))
file.write ('\n')
programs = categories[category]
c = 1
for (title, problem, solution) in sorted (programs):
file.write ('**' + str(c) + ') {0}** \n'.format (title))
file.write (' * [Problem Statement]({0}) \n'.format (problem))
file.write (' * [Solution]({0}) \n\n'.format (solution))
c+=1
file.write ('---\n')
if len (category_names) > 0:
file.write ('''
---
''')
#Create a README.md file with a nice index for using it directly from GitHub.
def create_README ():
category_names = get_dirs ()
count, categories = get_category_dict (category_names)
print_file (category_names, count, categories)
os.system('git add README.md')
if __name__ == '__main__':
create_README ()