This repository contains a simple command line utility used for calculating total size (both KB and lines of code) of program's code.
The main goal of this program is to calculate total size of a source code.
You need to specify the directory that contains the source code and file extension(s) of the desired files.
(for example html, css and js).
After that, the program searches the directory for files with the given file extension(s). It's also possible to exclude selected subdirectories from the search
(this is especially useful for directories like .venv, .git and so on).
The program typically prints:
- number of files found
- total number of lines in these files (newline at the end of the file isn't counted)
- total size of these files in KB (rounded to 2 decimal places). 1 KB = 1024 bytes
See the usage section for further details.
- Python version >= 3.9
- pip
Install via pip
pip install code-size-counterThis installs the program as a local python package and allows you to use code-size-counter CLI command.
See the PyPI page for release history.
Below, you can see list of all possible arguments (you can also get the help by running code-size-counter -h). directory is the only
one required.
usage: main.py [-h] [-d DIRECTORY] [-e EXTENSION [EXTENSION ...]] [-l] [-x EXCLUDE [EXCLUDE ...]] [-p {kb_size,lines,files}]
Calculate the total size (both KB and lines of code) of program's code.
options:
-h, --help show this help message and exit
-d DIRECTORY, --directory DIRECTORY
Path to the directory where to search files. The path can be either absolute or relative; leave empty if you want to search the current directory.
-e EXTENSION [EXTENSION ...], --extension EXTENSION [EXTENSION ...]
extensions of the files that we're searching (separated by spaces) for. Do not prefix them with a dot (e.g. use "py" instead of ".py"). Leave empty if you want to search for all files regardless of their extension.
-l, --log If present, the program prints its progress (e.g. 'file XXX processed')
-x EXCLUDE [EXCLUDE ...], --exclude EXCLUDE [EXCLUDE ...]
path to directories & files to exclude (separated by spaces). These paths are relative to the given directory (-d parameter)
-p {kb_size,lines,files}, --print {kb_size,lines,files}
Print just the selected value (KB size, total files or lines of code)
PS C:\Users\vojta\Programming\misc\code-size-counter>
The typical usage of this program looks like this (see below). In this case, we're looking for .py and .txt files in C:/Users/hacker123/Documents/hacking_app directory,
except those in .venv and .git subdirectories.
Note that we don't prefix the file extensions with a dot (e.g. use py instead of .py).
code-size-counter -d C:/Users/hacker123/Documents/hacking_app -e py txt -x .venv .gitPossible output
+---------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+---------------------------------------------------------+
| .py 1 50 1.51 |
| .txt 1 2 0.03 |
+---------------------------------------------------------+
| TOTAL 2 52 1.54 |
+---------------------------------------------------------+
See also other examples below.
Calculate the size of .py files inside ./tests/complex-test-dir directory.
code-size-counter -d ./tests/complex-test-dir -e pyOutput
+---------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+---------------------------------------------------------+
| .py 31 13030 462.61 |
+---------------------------------------------------------+
Calculate the size of .py files inside ./tests/complex-test-dir directory. Exclude virtualenv and src/module1 subdirectories.
code-size-counter -d ./tests/complex-test-dir -e py -x virtualenv src/module1Output
+---------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+---------------------------------------------------------+
| .py 8 268 8.26 |
+---------------------------------------------------------+
Calculate the size of .py files inside ./tests/complex-test-dir directory. Exclude virtualenv and src/module1 subdirectories and print logs.
code-size-counter -d ./tests/complex-test-dir -e py -x virtualenv src/module1 -lOutput
./tests/complex-test-dir/src/module2/main.py processed
./tests/complex-test-dir/src/module2/__init__.py processed
./tests/complex-test-dir/src/code_size_counter.py processed
./tests/complex-test-dir/src/file_tools.py processed
./tests/complex-test-dir/src/__init__.py processed
./tests/complex-test-dir/tests/test_code_size_counter.py processed
./tests/complex-test-dir/tests/__init__.py processed
./tests/complex-test-dir/code_size_counter.py processed
+---------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+---------------------------------------------------------+
| .py 8 268 8.26 |
+---------------------------------------------------------+
Calculate the size of .py, .md and .yml files inside ./tests/complex-test-dir directory. Exclude virtualenv and src/module1 subdirectories.
code-size-counter -d ./tests/complex-test-dir -e py md yml -x virtualenv src/module1Output
+---------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+---------------------------------------------------------+
| .yml 1 36 1.16 |
| .py 8 268 8.26 |
| .md 1 3 0.14 |
+---------------------------------------------------------+
| TOTAL 10 307 9.56 |
+---------------------------------------------------------+
Calculate the size of all files (regardless of their extension) inside ./directories/my-dir directory.
code-size-counter -d ./directories/my-dirPossible output
+----------------------------------------------------------+
| Extension Total files Total lines Total size (KB) |
+----------------------------------------------------------+
| (NONE) 1 10 0.11 |
| .flake8 1 2 0.03 |
| .gitignore 2 140 1.95 |
| .md 2 215 8.83 |
| .py 44 13533 477.44 |
| .yml 2 76 2.34 |
+----------------------------------------------------------+
| TOTAL 52 13976 490.70 |
+----------------------------------------------------------+
Also note the (NONE) extension, which means that the file has no extension
Calculate the size of .py files inside ./tests/complex-test-dir directory. Exclude virtualenv and src/module1 subdirectories
and print just the number of lines.
code-size-counter -d ./tests/complex-test-dir -e py -x virtualenv src/module1 -p linesOutput
268
You can then pipe the result (as you can see on the example below).
code-size-counter -d ./tests/complex-test-dir -e py -x virtualenv src/module1 -p lines | python -c "loc = input(); print(f'I copied {loc} lines from StackOverflow.')"Output
I copied 268 lines from StackOverflow.
We're using Poetry for dependency management and packaging.
- code_size_counter folder - source code files of the program
- main.py is the entry-point of the app
- tests folder - unit tests of the program (using
Python unittestmodule). The subfolders serve as a test data.
To run the tests, use the following command
poetry run python -m unittest discover tests -vRun this command in the root folder to execute main.py as a script.
poetry run python -m code_size_counter.main