можно в брачном контракте, можно просто зафиксировать в совместном документе, что на покупку квартиры одним супругом потрачено столько то средств полученных от продажи личного супруга, и второй супруг это признает.в случае развода и раздела имущества такой документ будет представлен для выделения дополнительной доли одному из супругов сверх половины положенной по закону