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
|
#include <Python.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <string.h>
#include <sys/stat.h>
static PyObject *sourcestamp_compute(PyObject *self, PyObject *args);
static PyMethodDef module_methods[] = {
{"compute", sourcestamp_compute, METH_VARARGS, NULL},
{NULL, NULL, 0, NULL}
};
PyMODINIT_FUNC
initsourcestamp(void)
{
Py_InitModule("sourcestamp", module_methods);
}
static void
list_dir(const char *dir, time_t *mtime, int *n_files)
{
DIR *dp;
struct dirent *entry;
struct stat statbuf;
dp = opendir(dir);
chdir(dir);
while ((entry = readdir(dp)) != NULL) {
*n_files = *n_files + 1;
lstat(entry->d_name, &statbuf);
if (statbuf.st_mtime > *mtime) {
*mtime = statbuf.st_mtime;
}
if(S_ISDIR(statbuf.st_mode)) {
if(strcmp(".", entry->d_name) == 0 ||
strcmp("..", entry->d_name) == 0 ||
strcmp(".git", entry->d_name) == 0)
continue;
list_dir(entry->d_name, mtime, n_files);
}
}
chdir("..");
closedir(dp);
}
static
PyObject *sourcestamp_compute(PyObject *self, PyObject *args)
{
time_t mtime = 0;
int n_files = 0;
const char *path;
char stamp[100];
if (!PyArg_ParseTuple(args, "s", &path)) {
return NULL;
}
list_dir(path, &mtime, &n_files);
snprintf(stamp, 100, "%d - %lu", n_files, (long unsigned)mtime);
return Py_BuildValue("s", stamp);
}
|
