Add a --limit command-line option which limits written Atom feeds.
Chris Pressey
7 years ago
19 | 19 | ) |
20 | 20 | argparser.add_argument('--output-atom', metavar='FILENAME', type=str, |
21 | 21 | help='Construct an Atom XML feed from the entries and write it out to this file' |
22 | ) | |
23 | argparser.add_argument('--limit', metavar='COUNT', type=int, default=None, | |
24 | help='Process no more than this many entries when making an Atom feed' | |
22 | 25 | ) |
23 | 26 | |
24 | 27 | options = argparser.parse_args(sys.argv[1:]) |
63 | 66 | write(u' {}'.format(entry_name)) |
64 | 67 | |
65 | 68 | if options.output_atom: |
66 | feedmark_atomize(documents, options.output_atom) | |
69 | feedmark_atomize(documents, options.output_atom, limit=options.limit) | |
67 | 70 | |
68 | 71 | |
69 | 72 | if __name__ == '__main__': |