Beispiel #1
0
PyObject *Vcf_writeFile( PyObject *self, PyObject *args ){
	char *filename;
	PyObject *cards;


	if(!(PyArg_ParseTuple( args, "sO", &filename, &cards)) ){
		return NULL;
	}

	VcFile saveFile;
	FILE *writeFile;

	/*Grab the number of cards and allocate that many Vcard pointers*/
	saveFile.ncards = PyList_Size(cards);
	saveFile.cardp = malloc(sizeof(Vcard*) * saveFile.ncards);
	assert(saveFile.cardp != NULL);

	/*Start loading up cards*/
	int i,j, numProp; 
	PyObject *cardListObj, *cardObj;
	for(i = 0; i < saveFile.ncards;i++){
		/*Grab the number of props in the card to allocate the Vcard and the VLA*/
		cardListObj = PyList_GetItem(cards,i);
		numProp = PyList_Size(cardListObj);
		saveFile.cardp[i] = malloc(sizeof(Vcard) + (sizeof(VcProp) * numProp));
		assert(saveFile.cardp[i] != NULL);
		saveFile.cardp[i]->nprops = numProp;


		/*Parse the tuples in the list*/
		for(j = 0; j < numProp; j++){
			cardObj = PyList_GetItem(cardListObj,j);

			saveFile.cardp[i]->prop[j].partype = NULL;
			saveFile.cardp[i]->prop[j].parval= NULL;
			saveFile.cardp[i]->prop[j].value = NULL;
			saveFile.cardp[i]->prop[j].hook = NULL;

			VcPname tempName;
			char *tempVal;
			char *tempType;
			char *tempValue;

			PyArg_ParseTuple(cardObj,"izzz", 
				&tempName, &tempType,
				&tempVal, &tempValue);

			saveFile.cardp[i]->prop[j].name = tempName;
			if(tempType != NULL){
				saveFile.cardp[i]->prop[j].partype = malloc(sizeof(char) * strlen(tempType) + 1);
				assert(saveFile.cardp[i]->prop[j].partype);
				strcpy(saveFile.cardp[i]->prop[j].partype,tempType);
			}

			if(tempVal != NULL){
				saveFile.cardp[i]->prop[j].parval = malloc(sizeof(char) * strlen(tempVal) + 1);
				assert(saveFile.cardp[i]->prop[j].parval);
				strcpy(saveFile.cardp[i]->prop[j].parval,tempVal);
			}

			if(tempValue != NULL){
				saveFile.cardp[i]->prop[j].value = malloc(sizeof(char) * strlen(tempValue) + 1);
				assert(saveFile.cardp[i]->prop[j].value);
				strcpy(saveFile.cardp[i]->prop[j].value,tempValue);
			}

		}

	}

	/*Try to open file for writing*/
	writeFile = fopen(filename,"w+");

	if(filename == NULL){
		return Py_BuildValue("s", strerror(errno));
	}


	/*Try to write to the file*/

	VcStatus status = writeVcFile(writeFile, &saveFile);
	fclose(writeFile);

	if(status.code != OK){
		freeVcFile(&saveFile);
		return Py_BuildValue("iii", status.code,status.linefrom,status.lineto);
	}

	freeVcFile(&saveFile);
	return Py_BuildValue("s","OK");
}
Beispiel #2
0
int main(int argc, char**argv){
    if (argc == 1){
        fprintf(stderr,"You must enter atleast 1 command.\n");
        return EXIT_FAILURE;
    }
    /*Read in file from STDIN*/
    int write = 0;
    VcStatus writeStatus, readStatus;
    VcFile input;
    readStatus = readVcFile(stdin,&input);
    if (readStatus.code != OK){
        fprintf(stderr,"Read Vcfile failed with error code %d lines %d to %d.\n",readStatus.code,readStatus.linefrom,readStatus.lineto);
        return EXIT_FAILURE;
    }
    /*Run the sepcified command*/
    if (argc > 1){
        if (strcmpML("-info",argv[1]) == 0){
            if (argc > 2) return EXIT_FAILURE;
            vcfInfo(stdout,&input);
        }
        else if (strcmpML("-sort",argv[1]) == 0){
            if (argc > 2) return EXIT_FAILURE;
            vcfSort(&input);
            writeStatus = writeVcFile(stdout,&input);
            write = 1;
        }
        else if (strcmpML("-canon",argv[1]) == 0){
            if (argc > 2) return EXIT_FAILURE;
            vcfCanon(&input);
            writeStatus = writeVcFile(stdout,&input);
            write = 1;
        }
        else if (strcmpML("-select",argv[1]) == 0){
            if (argc < 3){
                fprintf(stderr,"ERROR: You must enter which cards to select.\n");
            }
            else{
                char * which = strdup(argv[2]);
                assert(which);
                
                int check = scanWhich(which);
                if (check == -1){
                    fprintf(stderr,"ERROR: Invalid format for selection.");
                }
                else{
                    vcfSelect(&input,which);
                    writeStatus = writeVcFile(stdout,&input);
                    write = 1;
                }
                free(which);
            }
        }
        else{
            fprintf(stderr,"ERROR: Invalid options.\n");
            freeVcFile(&input);
            return EXIT_FAILURE;
        }
    }
    /*Print the error*/
    if ((write == 1) && writeStatus.code == IOERR){
        fprintf(stderr,"IOERR on linefrom: %d. to lineto: %d.\n",writeStatus.linefrom,writeStatus.lineto);
    }
    /*Free and return*/
    freeVcFile(&input);
    return EXIT_SUCCESS;
}